Parameter Adjustmens and Ensemble Model Implementation

util/config.py

@@ -1,6 +1,8 @@
 from ml_lib.utils.config import Config
-from models.binary_classifier import BinaryClassifier
-from models.bandwise_binary_classifier import BandwiseBinaryClassifier
+from models.conv_classifier import ConvClassifier
+from models.bandwise_conv_classifier import BandwiseConvClassifier
+from models.bandwise_conv_multihead_classifier import BandwiseConvMultiheadClassifier
+from models.ensemble import Ensemble
 
 
 class MConfig(Config):
@@ -8,5 +10,8 @@ class MConfig(Config):
 
     @property
     def _model_map(self):
-        return dict(BinaryClassifier=BinaryClassifier,
-                    BandwiseBinaryClassifier=BandwiseBinaryClassifier)
+        return dict(ConvClassifier=ConvClassifier,
+                    BandwiseConvClassifier=BandwiseConvClassifier,
+                    BandwiseConvMultiheadClassifier=BandwiseConvMultiheadClassifier,
+                    Ensemble=Ensemble,
+                    )