Merge pull request #7554 from initrv/add-catboost-tensorboard

Add tensorboard for catboost
This commit is contained in:
Robert Caulk
2022-10-10 21:03:45 +02:00
committed by GitHub
8 changed files with 27 additions and 7 deletions

View File

@@ -1,4 +1,5 @@
import logging
from pathlib import Path
from typing import Any, Dict
from catboost import CatBoostClassifier, Pool
@@ -31,8 +32,9 @@ class CatboostClassifier(BaseClassifierModel):
)
cbr = CatBoostClassifier(
allow_writing_files=False,
allow_writing_files=True,
loss_function='MultiClass',
train_dir=Path(dk.data_path / 'tensorboard'),
**self.model_training_parameters,
)

View File

@@ -1,4 +1,5 @@
import logging
from pathlib import Path
from typing import Any, Dict
from catboost import CatBoostRegressor, Pool
@@ -41,7 +42,8 @@ class CatboostRegressor(BaseRegressionModel):
init_model = self.get_init_model(dk.pair)
model = CatBoostRegressor(
allow_writing_files=False,
allow_writing_files=True,
train_dir=Path(dk.data_path / 'tensorboard'),
**self.model_training_parameters,
)

View File

@@ -1,4 +1,5 @@
import logging
from pathlib import Path
from typing import Any, Dict
from catboost import CatBoostRegressor, Pool
@@ -26,7 +27,8 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
"""
cbr = CatBoostRegressor(
allow_writing_files=False,
allow_writing_files=True,
train_dir=Path(dk.data_path / 'tensorboard'),
**self.model_training_parameters,
)