remove tensorboard dir from other pred models

This commit is contained in:
robcaulk 2022-10-11 19:49:24 +02:00
parent 5b5bb8aab5
commit dba1b573bc
2 changed files with 2 additions and 2 deletions

View File

@ -34,7 +34,7 @@ class CatboostClassifier(BaseClassifierModel):
cbr = CatBoostClassifier(
allow_writing_files=True,
loss_function='MultiClass',
train_dir=Path(dk.data_path / 'tensorboard'),
train_dir=Path(dk.data_path),
**self.model_training_parameters,
)

View File

@ -28,7 +28,7 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
cbr = CatBoostRegressor(
allow_writing_files=True,
train_dir=Path(dk.data_path / 'tensorboard'),
train_dir=Path(dk.data_path),
**self.model_training_parameters,
)