From dba1b573bc773eb952c0830e52b4bf85f64cddf0 Mon Sep 17 00:00:00 2001 From: robcaulk Date: Tue, 11 Oct 2022 19:49:24 +0200 Subject: [PATCH] remove tensorboard dir from other pred models --- freqtrade/freqai/prediction_models/CatboostClassifier.py | 2 +- .../freqai/prediction_models/CatboostRegressorMultiTarget.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/prediction_models/CatboostClassifier.py b/freqtrade/freqai/prediction_models/CatboostClassifier.py index 6dab012d1..2aebc3ebf 100644 --- a/freqtrade/freqai/prediction_models/CatboostClassifier.py +++ b/freqtrade/freqai/prediction_models/CatboostClassifier.py @@ -34,7 +34,7 @@ class CatboostClassifier(BaseClassifierModel): cbr = CatBoostClassifier( allow_writing_files=True, loss_function='MultiClass', - train_dir=Path(dk.data_path / 'tensorboard'), + train_dir=Path(dk.data_path), **self.model_training_parameters, ) diff --git a/freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py b/freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py index 085c13b86..de7a73e3a 100644 --- a/freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py +++ b/freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py @@ -28,7 +28,7 @@ class CatboostRegressorMultiTarget(BaseRegressionModel): cbr = CatBoostRegressor( allow_writing_files=True, - train_dir=Path(dk.data_path / 'tensorboard'), + train_dir=Path(dk.data_path), **self.model_training_parameters, )