kedro.extras.datasets.tensorflow

package documentation

(source)

Provides I/O for TensorFlow Models.

Module tensorflow_model_dataset TensorflowModelDataset is a data set implementation which can save and load TensorFlow models.

From __init__.py:

Class TensorFlowModelDataset TensorflowModelDataset loads and saves TensorFlow models. The underlying functionality is supported by, and passes input arguments through to, TensorFlow 2.X load_model and save_model methods.