Add сatboost train_dir for tensorboard

2022-10-06 19:59:35 +03:00
parent b0eff4160f
commit 86c781798a
3 changed files with 6 additions and 3 deletions
--- a/freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py
+++ b/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,
        )