steffen/ml_lib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Readme

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

12
utils/config.py
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: