tensorflow保存.h5模型的代码

以下是使用TensorFlow保存.h5模型的代码示例：

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense

# 构建模型
model = Sequential([
    Dense(units=64, activation='relu', input_shape=(32,)),
    Dense(units=10, activation='softmax')
])

# 编译模型
model.compile(optimizer=tf.keras.optimizers.Adam(0.001),
              loss='categorical_crossentropy',
              metrics=['accuracy'])

# 训练模型
model.fit(x_train, y_train, epochs=10, validation_data=(x_test, y_test))

# 保存模型为.h5文件
model.save('my_model.h5')

在上面的示例中，我们首先使用TensorFlow构建了一个简单的神经网络模型，然后编译并训练该模型。最后，我们调用`model.save('my_model.h5')`将模型保存为.h5文件。