dcase_models.model.ModelContainer¶
-
class
dcase_models.model.ModelContainer(model=None, model_path=None, model_name='ModelContainer', metrics=['classification'])[source]¶ Bases:
objectAbstract base class to store and manage models.
Parameters: - model : keras model or similar
Object that defines the model (i.e keras.models.Model)
- model_path : str
Path to the model file
- model_name : str
Model name
- metrics : list of str
List of metrics used for evaluation
-
__init__(model=None, model_path=None, model_name='ModelContainer', metrics=['classification'])[source]¶ Initialize ModelContainer
Parameters: - model : keras model or similar
Object that defines the model (i.e keras.models.Model)
- model_path : str
Path to the model file
- model_name : str
Model name
- metrics : list of str
List of metrics used for evaluation
Methods
__init__([model, model_path, model_name, …])Initialize ModelContainer build()Missing docstring here check_if_model_exists(folder, **kwargs)Missing docstring here evaluate(X_test, Y_test[, scaler])Missing docstring here get_available_intermediate_outputs()Missing docstring here get_intermediate_output(output_ix_name)Missing docstring here get_number_of_parameters()Missing docstring here load_model_from_json(folder, **kwargs)Missing docstring here load_model_weights(weights_folder)Missing docstring here save_model_json(folder)Missing docstring here save_model_weights(weights_folder)Missing docstring here train()Missing docstring here