我对以下行的以下代码有一些问题 new_model = load_model('124446.model', custom_objects=None,compile=True) 这是代码:
import tensorflow as tf
from tensorflow.keras.models import load_model
mnist = tf.keras.datasets.mnist
(x_train,y_train), (x_test,y_test) = mnist.load_data()
x_train = tf.keras.utils.normalize(x_train,axis=1)
x_test = tf.keras.utils.normalize(x_test,axis=1)
model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(128,activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(128,activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(10,activation=tf.nn.softmax))
model.compile(optimizer='adam',
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
model.fit(x_train,y_train,epochs=3)
tf.keras.models.save_model(model,'124446.model')
val_loss, val_acc = model.evaluate(x_test,y_test)
print(val_loss, val_acc)
new_model = load_model('124446.model', custom_objects=None, compile=True)
prediction = new_model.predict([x_test])
print(prediction)
Run Code Online (Sandbox Code Playgroud)
错误是:
回溯(最近一次调用最后):文件“C:/Users/TanveerIslam/PycharmProjects/DeepLearningPractice/1.py”,第32行,在new_model = load_model('124446.model',custom_objects = None,compile = True)文件“ C:\Users\TanveerIslam\PycharmProjects\DeepLearningPractice\venv\lib\site-packages\tensorflow\python\keras\engine\ saving.py”,第 262 行,在 load_model Sample_weight_mode=sample_weight_mode 中) 文件“C:\Users\TanveerIslam\ PycharmProjects\DeepLearningPractice\venv\lib\site-packages\tensorflow\python\training\checkpointable\base.py”,第 426 行,在 _method_wrapper 方法(self,*args,**kwargs)文件“C:\Users\TanveerIslam\ PycharmProjects\DeepLearningPractice\venv\lib\site-packages\tensorflow\python\keras\engine\training.py”,第 525 行,在编译指标中,self.output_names)
AttributeError:“顺序”对象没有属性“output_names”
那么任何人都可以给我蚂蚁解决方案吗? …