From b2fe2385d561e0169abed9a3d859e3e0af789ac2 Mon Sep 17 00:00:00 2001 From: zhupr Date: Tue, 1 Jun 2021 21:02:32 +0800 Subject: [PATCH] fix XGBoost predict error --- qlib/contrib/model/xgboost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/xgboost.py b/qlib/contrib/model/xgboost.py index 2a38f4fe19..300326143f 100755 --- a/qlib/contrib/model/xgboost.py +++ b/qlib/contrib/model/xgboost.py @@ -62,7 +62,7 @@ def predict(self, dataset: DatasetH, segment: Union[Text, slice] = "test"): if self.model is None: raise ValueError("model is not fitted yet!") x_test = dataset.prepare(segment, col_set="feature", data_key=DataHandlerLP.DK_I) - return pd.Series(self.model.predict(xgb.DMatrix(x_test.values)), index=x_test.index) + return pd.Series(self.model.predict(xgb.DMatrix(x_test)), index=x_test.index) def get_feature_importance(self, *args, **kwargs) -> pd.Series: """get feature importance