Debugging

lib/utils/config.py

@@ -5,6 +5,7 @@ from collections import defaultdict
 from configparser import ConfigParser
 from pathlib import Path
 
+from lib.models.homotopy_classification.cnn_based import ConvHomDetector
 from lib.utils.model_io import ModelParameters
 
 
@@ -24,6 +25,15 @@ class Config(ConfigParser):
     # for section in self.default_sections:
     #     self.__setattr__(section, property(lambda x :x._get_namespace_for_section(section))
 
+    @property
+    def model_class(self):
+        model_dict = dict(classifier_cnn=ConvHomDetector)
+        try:
+            return model_dict[self.get('model', 'type')]
+        except KeyError as e:
+            raise KeyError(rf'The model alias you provided ("{self.get("model", "type")}") does not exist! \n'
+                           f'Try one of these:\n{list(model_dict.keys())}')
+
     @property
     def main(self):
         return self._get_namespace_for_section('main')