isolate and standardize location of tensorboard files, add doc, ensure backtesting functionality

This commit is contained in:
robcaulk
2022-10-09 21:11:37 +02:00
parent ec7af83c87
commit a4aa1b972c
6 changed files with 23 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
@@ -34,7 +35,7 @@ class CatboostClassifier(BaseClassifierModel):
cbr = CatBoostClassifier(
allow_writing_files=True,
loss_function='MultiClass',
train_dir=dk.data_path,
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
@@ -42,7 +43,7 @@ class CatboostRegressor(BaseRegressionModel):
model = CatBoostRegressor(
allow_writing_files=True,
train_dir=dk.data_path,
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
@@ -27,7 +28,7 @@ class CatboostRegressorMultiTarget(BaseRegressionModel):
cbr = CatBoostRegressor(
allow_writing_files=True,
train_dir=dk.data_path,
train_dir=Path(dk.data_path / 'tensorboard'),
**self.model_training_parameters,
)