Python tf.compat.v1.keras.experimental.load_from_saved_model用法及代碼示例

從 export_saved_model() 創建的 SavedModel 加載 keras 模型。

用法

tf.compat.v1.keras.experimental.load_from_saved_model(
    saved_model_path, custom_objects=None
)

此函數通過以下方式重新實例化模型狀態：

1)從json加載模型拓撲(這最終將來自metagraph)。 2)從檢查點加載模型權重。

例子：

import tensorflow as tf

# Create a tf.keras model.
model = tf.keras.Sequential()
model.add(tf.keras.layers.Dense(1, input_shape=[10]))
model.summary()

# Save the tf.keras model in the SavedModel format.
path = '/tmp/simple_keras_model'
tf.keras.experimental.export_saved_model(model, path)

# Load the saved keras model back.
new_model = tf.keras.experimental.load_from_saved_model(path)
new_model.summary()