`neural_compressor.model.keras_model`¶

Class for Keras model.

Module Contents¶

Classes¶

Build Keras model.

class neural_compressor.model.keras_model.KerasModel(model, **kwargs)¶

Bases: neural_compressor.model.base_model.BaseModel

Build Keras model.

property q_config¶: Return q_config.

property model¶: Return model itself.

property graph_info¶: Return graph info.

abstract save(root, *args, **kwargs)¶: Save Keras model.

abstract framework()¶: Return framework.