liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 16:26:55 +08:00
Code Activity

Merge pull request #457 from zhupr/fix_XGBoost_predict_error

Browse Source 
fix XGBoost predict error
This commit is contained in:
you-n-g
2021-06-02 16:14:18 +08:00
committed by GitHub
parent 4ff0c4fb0f b2fe2385d5
commit 64896745d0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/xgboost.py
View File

@@ -62,7 +62,7 @@ class XGBModel(Model, FeatureInt):
if self.model is None: if self.model is None:
raise ValueError("model is not fitted yet!") raise ValueError("model is not fitted yet!")
x_test = dataset.prepare(segment, col_set="feature", data_key=DataHandlerLP.DK_I) 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: def get_feature_importance(self, *args, **kwargs) -> pd.Series:
"""get feature importance """get feature importance