code_5.py 201 字节
Newer Older
ToTensor's avatar
ToTensor 已提交
1 2 3 4 5
plt.plot(loss_history, 'g--', label='Loss Curve') # 显示损失曲线
plt.xlabel('Iterations') # x轴标签
plt.ylabel('Loss') # y轴标签
plt.legend() # 显示图例
plt.show() # 显示损失曲线