catboost tensorboard bugfix

This commit is contained in:
robcaulk 2022-10-11 19:05:46 +02:00
parent 8f2a887a58
commit 5b5bb8aab5
3 changed files with 2 additions and 3 deletions

View File

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

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

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