Skip to content

Commit

Permalink
Support Reweighter for HighFreq Model (#908)
Browse files Browse the repository at this point in the history
  • Loading branch information
you-n-g authored Feb 7, 2022
1 parent d496cf7 commit 3274e16
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions qlib/contrib/model/highfreq_gdbt_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,6 @@ def fit(
early_stopping_rounds=50,
verbose_eval=20,
evals_result=dict(),
**kwargs
):
dtrain, dvalid = self._prepare_data(dataset)
self.model = lgb.train(
Expand All @@ -123,7 +122,6 @@ def fit(
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]
Expand Down

0 comments on commit 3274e16

Please sign in to comment.