Add lightgbm classifier, add classifier check test, fix classifier bug.

2022-08-06 17:51:21 +02:00
parent 47a30047eb
commit eb8bde37c1
8 changed files with 249 additions and 6 deletions
--- a/freqtrade/freqai/prediction_models/CatboostClassifier.py
+++ b/freqtrade/freqai/prediction_models/CatboostClassifier.py
@@ -32,9 +32,6 @@ class CatboostClassifier(BaseRegressionModel):

        cbr = CatBoostClassifier(
            allow_writing_files=False,
-            gpu_ram_part=0.5,
-            verbose=100,
-            early_stopping_rounds=400,
            loss_function='MultiClass',
            **self.model_training_parameters,
        )