pimlico.datatypes.keras module

Datatypes for storing and loading Keras models.

class pimlico.datatypes.keras.KerasModel(base_dir, pipeline, additional_name=None, use_main_metadata=False, **kwargs)

Bases: pimlico.datatypes.base.PimlicoDatatype

Datatype for both types of Keras models, stored using Keras’ own storage mechanisms.

get_custom_objects()

get_software_dependencies()

load_model()

custom_objects = {}

class pimlico.datatypes.keras.KerasModelWriter(base_dir, **kwargs)

Bases: pimlico.datatypes.base.PimlicoDatatypeWriter

Writer for storing both types of Keras model (since they provide the same storage interface).

write_architecture(model)

write_model(model)

write_weights(model)