Readme
This commit is contained in:
@ -9,6 +9,9 @@ Clone it to find a collection of:
|
|||||||
- Utility Function for Model I/O
|
- Utility Function for Model I/O
|
||||||
- DL Modules
|
- DL Modules
|
||||||
- A Plotter Object
|
- A Plotter Object
|
||||||
|
- Audio Related Tools and Funtion
|
||||||
|
- Librosa
|
||||||
|
- Scipy Signal
|
||||||
|
|
||||||
###Notes:
|
###Notes:
|
||||||
- Use directory links to link from your main project folder to the ml_lib folder. Pycharm will automatically use
|
- 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
|
@property
|
||||||
def model_class(self):
|
def model_class(self):
|
||||||
model_dict = dict(ConvHomDetector=ConvHomDetector,
|
model_dict = dict(BinaryClassifier=BinaryClassifier,
|
||||||
CNNRouteGenerator=CNNRouteGeneratorModel,
|
|
||||||
CNNRouteGeneratorDiscriminated=CNNRouteGeneratorDiscriminated
|
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
return model_dict[self.get('model', 'type')]
|
return model_dict[self.get('model', 'type')]
|
||||||
@ -106,6 +104,14 @@ class Config(ConfigParser):
|
|||||||
new_config.read_dict(sorted_dict)
|
new_config.read_dict(sorted_dict)
|
||||||
return new_config
|
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):
|
def update(self, mapping):
|
||||||
sorted_dict = self._sort_combined_section_key_mapping(mapping)
|
sorted_dict = self._sort_combined_section_key_mapping(mapping)
|
||||||
for section in sorted_dict:
|
for section in sorted_dict:
|
||||||
|
Reference in New Issue
Block a user