project Refactor, CNN Classifier Basics

2020-03-08 23:46:02 +01:00
parent 75e8a61628
commit cd4fdf2de3
20 changed files with 441 additions and 239 deletions
@@ -5,6 +5,7 @@ from collections import defaultdict
 from configparser import ConfigParser
 from pathlib import Path

+from lib.models.generators.cnn import CNNRouteGeneratorModel
 from lib.models.homotopy_classification.cnn_based import ConvHomDetector
 from lib.utils.model_io import ModelParameters

@@ -27,7 +28,7 @@ class Config(ConfigParser):

    @property
    def model_class(self):
-        model_dict = dict(classifier_cnn=ConvHomDetector)
+        model_dict = dict(classifier_cnn=ConvHomDetector, generator_cnn=CNNRouteGeneratorModel)
        try:
            return model_dict[self.get('model', 'type')]
        except KeyError as e: