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

minor fix

Browse Source
This commit is contained in:
Haoyu Wang
2020-11-06 17:47:43 +08:00
parent 1556be6798
commit eead71fcb5
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
qlib/contrib/model/catboost_model.py
View File

@@ -54,11 +54,13 @@ class CatBoostModel(Model):
self.model.fit( self.model.fit(
train_pool, train_pool,
eval_set = valid_pool, eval_set = valid_pool,
use_best_model = True use_best_model = True,
**kwargs
) )
evals_result["train"] = list(self.model.get_evals_result().values())[0] evals_result = self.model.get_evals_result()
evals_result["valid"] = self.model.get_test_eval() evals_result["train"] = list(evals_result["learn"].values())[0]
evals_result["valid"] = list(evals_result["validation"].values())[0]
def predict(self, dataset): def predict(self, dataset):