isolate and standardize location of tensorboard files, add doc, ensure backtesting functionality
This commit is contained in:
@@ -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,
|
||||
)
|
||||
|
||||
|
@@ -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,
|
||||
)
|
||||
|
||||
|
@@ -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,
|
||||
)
|
||||
|
||||
|
Reference in New Issue
Block a user