All LGBMRegressor model parameters are now set in config

2022-07-03 16:30:01 +02:00 · 2022-07-03 16:30:01 +02:00 · 8ac8d53c32
commit 8ac8d53c32
parent a6077ac7f4
1 changed files with 1 additions and 1 deletions
--- a/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py
+++ b/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py
@ -85,7 +85,7 @@ class LightGBMPredictionModel(IFreqaiModel):
        X = data_dictionary["train_features"]
        y = data_dictionary["train_labels"]

-        model = LGBMRegressor(seed=42, n_estimators=2000, verbosity=1, force_col_wise=True)
+        model = LGBMRegressor(**self.model_training_parameters)
        model.fit(X=X, y=y, eval_set=eval_set)

        return model