add exception for not passing timerange. Remove hard coded arguments for CatboostPredictionModels. Update docs

2022-07-24 09:01:23 +02:00
parent fff39eff9e
commit 88e10f7306
4 changed files with 4 additions and 6 deletions
--- a/freqtrade/freqai/prediction_models/CatboostPredictionModel.py
+++ b/freqtrade/freqai/prediction_models/CatboostPredictionModel.py
@@ -38,8 +38,6 @@ class CatboostPredictionModel(BaseRegressionModel):

        model = CatBoostRegressor(
            allow_writing_files=False,
-            verbose=100,
-            early_stopping_rounds=400,
            **self.model_training_parameters,
        )
        model.fit(X=train_data, eval_set=test_data)
--- a/freqtrade/freqai/prediction_models/CatboostPredictionMultiModel.py
+++ b/freqtrade/freqai/prediction_models/CatboostPredictionMultiModel.py
@@ -27,9 +27,6 @@ class CatboostPredictionMultiModel(BaseRegressionModel):

        cbr = CatBoostRegressor(
            allow_writing_files=False,
-            gpu_ram_part=0.5,
-            verbose=100,
-            early_stopping_rounds=400,
            **self.model_training_parameters,
        )