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

Merge pull request #7576 from freqtrade/bugfix-tensorboard

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/data_kitchen.py
View File

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

2
freqtrade/freqai/freqai_interface.py
View File

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

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/CatboostRegressor.py
View File

@ -43,7 +43,7 @@ class CatboostRegressor(BaseRegressionModel):
model = CatBoostRegressor( model = 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,
) )

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