13 lines
408 B
Python
13 lines
408 B
Python
from ml_lib.utils.config import Config
|
|
from models.binary_classifier import BinaryClassifier
|
|
from models.bandwise_binary_classifier import BandwiseBinaryClassifier
|
|
|
|
|
|
class MConfig(Config):
|
|
# TODO: There should be a way to automate this.
|
|
|
|
@property
|
|
def _model_map(self):
|
|
return dict(BinaryClassifier=BinaryClassifier,
|
|
BandwiseBinaryClassifier=BandwiseBinaryClassifier)
|