【发布时间】:2019-03-10 20:56:09
【问题描述】:
下面的代码有问题 以下行的 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)
错误是:
Traceback(最近一次调用最后一次):文件 "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 中 方法(自我,*args,**kwargs)文件“C:\Users\TanveerIslam\PycharmProjects\DeepLearningPractice\venv\lib\site-packages\tensorflow\python\keras\engine\training.py”, 第 525 行,在编译中 指标,self.output_names)
AttributeError: 'Sequential' 对象没有属性 'output_names'
那么任何人都可以给我蚂蚁解决方案。
注意:我使用 pycharm 作为 IDE。
【问题讨论】:
-
代码运行正常。我不知道是什么问题。也许尝试指定保存/加载文件的某个位置。
-
感谢您的回复。但我也使用了文件(模型)的真实路径。但显示相同的错误“AttributeError:'Sequential'对象没有属性'output_names'”并且我使用了pycharm IDE
标签: python tensorflow machine-learning keras deep-learning