adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

All LGBMRegressor model parameters are now set in config

Browse Source
This commit is contained in:
robcaulk 2022-07-03 16:30:01 +02:00
parent a6077ac7f4
commit 8ac8d53c32
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/prediction_models/LightGBMPredictionModel.py
View File

@ -85,7 +85,7 @@ class LightGBMPredictionModel(IFreqaiModel):
X = data_dictionary["train_features"] X = data_dictionary["train_features"]
y = data_dictionary["train_labels"] 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) model.fit(X=X, y=y, eval_set=eval_set)
return model return model