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

Add сatboost train_dir for tensorboard

Browse Source
This commit is contained in:
initrv
2022-10-06 19:59:35 +03:00
parent b0eff4160f
commit 86c781798a
3 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -32,8 +32,9 @@ class CatboostClassifier(BaseClassifierModel):
) )
cbr = CatBoostClassifier( cbr = CatBoostClassifier(
allow_writing_files=False, allow_writing_files=True,
loss_function='MultiClass', loss_function='MultiClass',
train_dir=dk.data_path,
**self.model_training_parameters, **self.model_training_parameters,
) )

3
freqtrade/freqai/prediction_models/CatboostRegressor.py
View File

@@ -41,7 +41,8 @@ class CatboostRegressor(BaseRegressionModel):
init_model = self.get_init_model(dk.pair) init_model = self.get_init_model(dk.pair)
model = CatBoostRegressor( model = CatBoostRegressor(
allow_writing_files=False, allow_writing_files=True,
train_dir=dk.data_path,
**self.model_training_parameters, **self.model_training_parameters,
) )

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

@@ -26,7 +26,8 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
""" """
cbr = CatBoostRegressor( cbr = CatBoostRegressor(
allow_writing_files=False, allow_writing_files=True,
train_dir=dk.data_path,
**self.model_training_parameters, **self.model_training_parameters,
) )