pytorch画acc和loss曲线
时间: 2023-09-08 21:15:21 浏览: 96
pytorch绘制并显示loss曲线和acc曲线,LeNet5识别图像准确率
5星 · 资源好评率100%
好的,关于这个问题,您可以使用matplotlib库绘制acc和loss曲线,具体操作可以参考类似于以下的代码:
```
import matplotlib.pyplot as plt
# 训练过程中的acc和loss数据
train_loss = [0.1, 0.2, 0.15, 0.08, 0.05]
train_acc = [0.9, 0.8, 0.85, 0.92, 0.95]
# 定义绘图函数
def plot_curve(x,y,title):
plt.plot(x,y)
plt.title(title)
plt.xlabel('Epoch')
plt.ylabel('Value')
plt.show()
# 绘制训练过程中的loss曲线
plot_curve(range(1,6), train_loss, 'Train Loss')
# 绘制训练过程中的acc曲线
plot_curve(range(1,6), train_acc, 'Train Acc')
```
希望以上代码能够帮助您绘制出您所需要的acc和loss曲线。
阅读全文