Readme
This commit is contained in:
parent
b6d76df71f
commit
5c036e6c11
@ -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
|
||||
|
@ -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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user