1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-06 05:51:17 +08:00

Support Reweighter for HighFreq Model (#908)

This commit is contained in:
you-n-g
2022-02-07 21:45:53 +08:00
committed by GitHub
parent d496cf7476
commit 3274e16c95

View File

@@ -111,7 +111,6 @@ class HFLGBModel(ModelFT, LightGBMFInt):
early_stopping_rounds=50,
verbose_eval=20,
evals_result=dict(),
**kwargs
):
dtrain, dvalid = self._prepare_data(dataset)
self.model = lgb.train(
@@ -123,7 +122,6 @@ class HFLGBModel(ModelFT, LightGBMFInt):
early_stopping_rounds=early_stopping_rounds,
verbose_eval=verbose_eval,
evals_result=evals_result,
**kwargs
)
evals_result["train"] = list(evals_result["train"].values())[0]
evals_result["valid"] = list(evals_result["valid"].values())[0]