Re-add early stopping during final fit

ThomasMeissnerDS · Jul 2, 2023 · 3c16654 · 3c16654
1 parent 3d7cc4d
commit 3c16654
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/bluecast/ml_modelling/xgboost.py b/bluecast/ml_modelling/xgboost.py
@@ -131,7 +131,7 @@ def fit(
                 self.conf_params_xgboost.params,
                 d_train,
                 num_boost_round=self.conf_params_xgboost.params["steps"],
-                # early_stopping_rounds=self.conf_training.early_stopping_rounds,
+                early_stopping_rounds=self.conf_training.early_stopping_rounds,
                 evals=eval_set,
             )
         print("Finished training")

diff --git a/dist/bluecast-0.8-py3-none-any.whl b/dist/bluecast-0.8-py3-none-any.whl
diff --git a/dist/bluecast-0.8.tar.gz b/dist/bluecast-0.8.tar.gz