BandwiseBinaryClassifier is work in progress; TODO: Shape Piping.

2020-05-04 18:45:13 +02:00
parent e4f6506a4b
commit 451f78f820
7 changed files with 190 additions and 42 deletions
--- a/util/config.py
+++ b/util/config.py
@ -1,9 +1,12 @@
 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)
+        return dict(BinaryClassifier=BinaryClassifier,
+                    BandwiseBinaryClassifier=BandwiseBinaryClassifier)