adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove tensorboard dir from other pred models

Browse Source
This commit is contained in:
robcaulk
2022-10-11 19:49:24 +02:00
parent 5b5bb8aab5
commit dba1b573bc
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/prediction_models/CatboostClassifier.py
View File

@@ -34,7 +34,7 @@ class CatboostClassifier(BaseClassifierModel):
cbr = CatBoostClassifier( cbr = CatBoostClassifier(
allow_writing_files=True, allow_writing_files=True,
loss_function='MultiClass', loss_function='MultiClass',
train_dir=Path(dk.data_path / 'tensorboard'), train_dir=Path(dk.data_path),
**self.model_training_parameters, **self.model_training_parameters,
) )

2
freqtrade/freqai/prediction_models/CatboostRegressorMultiTarget.py
View File

@@ -28,7 +28,7 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
cbr = CatBoostRegressor( cbr = CatBoostRegressor(
allow_writing_files=True, allow_writing_files=True,
train_dir=Path(dk.data_path / 'tensorboard'), train_dir=Path(dk.data_path),
**self.model_training_parameters, **self.model_training_parameters,
) )