import matplotlib.pyplot as plt
loss_list = []
accuracy_list = []
def main():
loss_list = [1, 2, 3, 4]
accuracy_list = [3, 2, 1, 4]
x1 = range(0, 4)
x2 = range(0, 4)
adam_lr=0.001
epochs=400
y1 = accuracy_list
y2 = loss_list
plt.subplot(2, 1, 1)
plt.plot(x1, y1, 'o-')
plt.title('acc vs loss;lr={0},epochs={1}'.format(adam_lr, epochs))
plt.subplot(2, 1, 2)
plt.plot(x2, y2, '.-')
plt.xlabel('loss,acc')
plt.savefig('./result_vgg.jpg')
plt.show()
if __name__ == '__main__':
main()
在神经网络代码里,使用loss_list.append(x)记录下每次的loss即可。