Add сatboost train_dir for tensorboard

freqtrade/freqai/prediction_models/CatboostClassifier.py

@@ -32,8 +32,9 @@ class CatboostClassifier(BaseClassifierModel):
         )
 
         cbr = CatBoostClassifier(
-            allow_writing_files=False,
+            allow_writing_files=True,
             loss_function='MultiClass',
+            train_dir=dk.data_path,
             **self.model_training_parameters,
         )
 

freqtrade/freqai/prediction_models/CatboostRegressor.py

@@ -41,7 +41,8 @@ class CatboostRegressor(BaseRegressionModel):
         init_model = self.get_init_model(dk.pair)
 
         model = CatBoostRegressor(
-            allow_writing_files=False,
+            allow_writing_files=True,
+            train_dir=dk.data_path,
             **self.model_training_parameters,
         )
 

freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py

@@ -26,7 +26,8 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
         """
 
         cbr = CatBoostRegressor(
-            allow_writing_files=False,
+            allow_writing_files=True,
+            train_dir=dk.data_path,
             **self.model_training_parameters,
         )