报错“too many indices for array: array is 1-dimensional, but 2 were indexed”，数据和代码都是根据别人发表的文章共享的，不知道问题出在哪
initial training of underlying models...
CV.. 1/10
---------------------------------------------------------------------------
IndexError Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_15028/761714535.py in <module>
1 model_sq = SurvivalQuilts()
----> 2 model_sq.train(tr_X, tr_T, tr_Y)
D:\python file\SurvivalQuilts-master\class_SurvivalQuilts.py in train(self, X, T, Y)
55 for cv_idx in range(self.num_cv):
56 print('CV.. {}/{}'.format(cv_idx+1, self.num_cv))
---> 57 pulled_models, tmp_CINDEX, tmp_BRIER = self._get_models_pulled_CV(X, T, Y, seed=cv_idx)
58
59 metric_CINDEX[cv_idx,:,:] = tmp_CINDEX
D:\python file\SurvivalQuilts-master\class_SurvivalQuilts.py in _get_models_pulled_CV(self, X, T, Y, seed)
287
288 for m, model in enumerate(pulled_models):
--> 289 pred = model.predict(X_va, self.time_horizons)
290
291 for t, eval_time in enumerate(self.time_horizons):
D:\python file\SurvivalQuilts-master\class_UnderlyingModels.py in predict(self, X, time_horizons)
51 preds_[:, t] = 1. - surv[:, 0]
52 else:
---> 53 preds_[:, t] = 1. - surv[:, tmp_time[0]]
54
55 else:
IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed