Merge pull request #7576 from freqtrade/bugfix-tensorboard

catboost tensorboard bugfix
This commit is contained in:
Robert Caulk 2022-10-11 21:27:42 +02:00 committed by GitHub
commit 8ab600f7b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 4 additions and 6 deletions

View File

@ -119,8 +119,6 @@ class FreqaiDataKitchen:
/ f"sub-train-{pair.split('/')[0]}_{trained_timestamp}"
)
Path(self.data_path / 'tensorboard').mkdir(parents=True, exist_ok=True)
return
def make_train_test_datasets(

View File

@ -201,10 +201,10 @@ class IFreqaiModel(ABC):
new_trained_timerange,
data_load_timerange,
) = dk.check_if_new_training_required(trained_timestamp)
dk.set_paths(pair, new_trained_timerange.stopts)
if retrain:
self.train_timer('start')
dk.set_paths(pair, new_trained_timerange.stopts)
try:
self.extract_data_and_train_model(
new_trained_timerange, pair, strategy, dk, data_load_timerange

View File

@ -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,
)

View File

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

View File

@ -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,
)