unox.model

Functions

load_model(model_path)

Load a trained model from a file.

Module Contents

unox.model.load_model(model_path: str)[source]

Load a trained model from a file.

Parameters:

model_path (str) – Path to the model file. Should be an .h5 or .keras file.

Returns:

model – The loaded model.

Return type:

tf.keras.Model

Examples

>>> model = load_model('model.h5')
>>> model.summary()