训练:
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
history = model.fit(X, Y, validation_split=0.33, epochs=150, batch_size=10, verbose=0)
保存训练的历史记录
import pickle
with open('log.txt', 'wb') as file_txt:
pickle.dump(history.history, file_txt)
读取保存的训练历史记录:
with open('log.txt','rb') as file_txt:
history=pickle.load(file_txt)
绘制曲线
下面是绘制准确率的变化曲线,如果将acc修改成loss就可以绘制损失变化曲线。
plt.plot(history['acc'])
plt.plot(history['val_acc'])
plt.title("model accuracy")
plt.ylabel("Accuracy")
plt.xlabel("epoch")
plt.legend(["train","test"],loc="lower right")
plt.show()
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)