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
Show Stats 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(
allow_writing_files=False,
allow_writing_files=True,
loss_function='MultiClass',
train_dir=dk.data_path,
**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)
model = CatBoostRegressor(
allow_writing_files=False,
allow_writing_files=True,
train_dir=dk.data_path,
**self.model_training_parameters,
)

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

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