diff --git a/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py b/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py index 7882aca4a..cc39cec47 100644 --- a/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py +++ b/user_data/freqaimodels/CatboostPredictionBinaryMultiModel.py @@ -30,7 +30,7 @@ class CatboostPredictionBinaryMultiModel(BaseClassifierModel): cbr = CatBoostClassifier( allow_writing_files=False, - gpu_ram_part=0.5, + gpu_ram_part=0.2, verbose=100, early_stopping_rounds=400, **self.model_training_parameters,