角膜预测模型损失或准确性的图表

Question

因此，我尝试为模型绘制一个图，说我有20个历元，并且该图应显示每个历元的准确性/损失。到目前为止，我已经在Keras网站上找到了此代码。

history = model.fit(x_train, y_train, epochs = 30, batch_size = 128,validation_split = 0.2)
plot(history)

我尝试将其用于数据。

import matplotlib.pyplot as plt
plt.plot(history)

这是我得到的错误

TypeError: float() argument must be a string or a number, not 'History'

是否有任何方法可以对此进行校正或为每个时期绘制图形？ 谢谢。

Answer 1

model_history = model.fit(...

plt.figure()
plt.subplot(211)
plt.plot(model_history.history['accuracy'])
plt.subplot(212)
plt.plot(model_history.history['loss'])

Answer 2

这段代码对我有用。

print(history.history.keys()) # Displays keys from history, in my case loss,acc
plt.plot(history.history['acc']) #here I am trying to plot only accuracy, the same can be used for loss as well
plt.title('model accuracy')
plt.ylabel('accuracy')
plt.xlabel('epoch')
plt.show()