This commit is contained in:
Si11ium 2020-04-15 15:59:51 +02:00
parent b6d76df71f
commit 5c036e6c11
2 changed files with 12 additions and 3 deletions

View File

@ -9,6 +9,9 @@ Clone it to find a collection of:
- Utility Function for Model I/O
- DL Modules
- A Plotter Object
- Audio Related Tools and Funtion
- Librosa
- Scipy Signal
###Notes:
- Use directory links to link from your main project folder to the ml_lib folder. Pycharm will automatically use

View File

@ -31,9 +31,7 @@ class Config(ConfigParser):
@property
def model_class(self):
model_dict = dict(ConvHomDetector=ConvHomDetector,
CNNRouteGenerator=CNNRouteGeneratorModel,
CNNRouteGeneratorDiscriminated=CNNRouteGeneratorDiscriminated
model_dict = dict(BinaryClassifier=BinaryClassifier,
)
try:
return model_dict[self.get('model', 'type')]
@ -106,6 +104,14 @@ class Config(ConfigParser):
new_config.read_dict(sorted_dict)
return new_config
def build_model(self):
return self.model_class(self.model_paramters)
def build_and_init_model(self, weight_init_function):
model = self.build_model()
model.init_weights(weight_init_function)
return model
def update(self, mapping):
sorted_dict = self._sort_combined_section_key_mapping(mapping)
for section in sorted_dict: