Model Training

This commit is contained in:
Si11ium
2020-05-03 18:00:51 +02:00
parent 8a97f59906
commit e4f6506a4b
9 changed files with 167 additions and 105 deletions

View File

@ -5,5 +5,5 @@ from models.binary_classifier import BinaryClassifier
class MConfig(Config):
@property
def model_map(self):
def _model_map(self):
return dict(BinaryClassifier=BinaryClassifier)

11
util/logging.py Normal file
View File

@ -0,0 +1,11 @@
from pathlib import Path
from ml_lib.utils.logging import Logger
class MLogger(Logger):
@property
def outpath(self):
# FIXME: Specify a special path
return Path(self.config.train.outpath)