在tensorflow.keras.experimental模型中,可以通过tf.saved_model.save
函数来导出模型。具体来说,可以通过以下步骤找到export_saved_model属性:
import tensorflow as tf
from tensorflow.keras import layers
model = tf.keras.Sequential([
layers.Dense(64, activation='relu', input_shape=(784,)),
layers.Dense(10, activation='softmax')
])
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
model.fit(x_train, y_train, epochs=10)
export_path = '/path/to/exported/model'
tf.saved_model.save(model, export_path)
在上述代码中,export_saved_model
属性即为tf.saved_model.save
函数,它用于将模型保存到指定路径。导出的模型可以用于后续的推理或部署。
腾讯云相关产品和产品介绍链接地址:
云+社区沙龙online第5期[架构演进]
云+社区技术沙龙[第19期]
Elastic 实战工作坊
Elastic 实战工作坊
DB TALK 技术分享会
北极星训练营
云+社区开发者大会 武汉站
云+社区技术沙龙[第10期]
领取专属 10元无门槛券
手把手带您无忧上云