无法以保存模型格式保存 张量流

Question

我正在尝试使用SavedModel格式保存我的ANN模型。我使用的命令是：

model.save("my_model")

它应该给我一个文件夹，即"；my_model"；它包含所有saved_model.pb、变量和资产，相反，它给了我一个HDF文件，即my_model。我使用keras v.2.3.1和tensorflow v.2.2.0

这是我的一点代码：

from keras import optimizers
from keras import backend
from keras.models import Sequential
from keras.layers import Dense
from keras.activations import relu,tanh,sigmoid
network_layout = []
for i in range(3):
network_layout.append(8)
model = Sequential()
#Adding input layer and first hidden layer
model.add(Dense(network_layout[0],  
name = "Input",
input_dim=inputdim,
kernel_initializer='he_normal',
activation=activation))
#Adding the rest of hidden layer
for numneurons in network_layout[1:]:
model.add(Dense(numneurons,
kernel_initializer = 'he_normal',
activation=activation))
#Adding the output layer
model.add(Dense(outputdim,
name="Output",
kernel_initializer="he_normal",
activation="relu"))
#Compiling the model
model.compile(optimizer=opt,loss='mse',metrics=['mse','mae','mape'])
model.summary()
#Training the model
history = model.fit(x=Xtrain,y=ytrain,validation_data=(Xtest,ytest),batch_size=32,epochs=epochs)
model.save('my_model')

我已经阅读了tensorflow网站上的API文档，并按照它的说法使用了没有任何文件扩展名的model.save("my_model"(，但我无法正确理解。

我们将非常感谢你的帮助。谢谢大家！

Answer 1

如果要使用tensorflow保存的模型格式，请使用：

tms_model = tf.saved_model.save(model,"export/1")

这将创建一个文件夹导出和其中的子文件夹1。在1文件夹中，您可以看到资产、变量和.pb文件。

希望这能帮助你。

确保像这样更改您的进口from tensorflow.keras import optimizers