Edit on GitHub

tensorflow

class TFKerasModel

MlemABC parent type: model_type

MlemABC type: tf_keras

:class:`.ModelType` implementation for Tensorflow Keras models

Fields:

  • io: ModelIO = TFKerasModelIO() - IO

class TFKerasModelIO

MlemABC parent type: model_io

MlemABC type: tf_keras

IO for Tensorflow Keras models (:class:`tensorflow.keras.Model`
objects)

Fields:

  • save_format: str - tf for custom net classes and h5 otherwise
Content

šŸ› Found an issue? Let us know! Or fix it:

Edit on GitHub

ā“ Have a question? Join our chat, we will help you:

Discord Chat