catboost tensorboard bugfix

This commit is contained in:
robcaulk
2022-10-11 19:05:46 +02:00
parent 8f2a887a58
commit 5b5bb8aab5
3 changed files with 2 additions and 3 deletions

View File

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