基于matplotlib绘制test loss折线图代码
时间: 2023-07-17 14:53:42 浏览: 40
以下是一个使用 Matplotlib 绘制测试集上 Loss 折线图的 Python 代码示例:
```python
import matplotlib.pyplot as plt
# 定义测试集的 Loss 值
test_loss = [1.2580, 1.0588, 0.9649, 0.7881, 0.7288, 0.6999, 0.6862, 0.6809, 0.6776, 0.6754, 0.6734, 0.6704, 0.6683, 0.6668, 0.6476, 0.6442, 0.6431, 0.6423, 0.6418, 0.6411, 0.6404, 0.6399, 0.6396, 0.6393, 0.6392, 0.6390, 0.6388, 0.6386, 0.6385, 1.1399]
# 定义 x 轴的取值范围
x = range(len(test_loss))
# 绘制折线图
plt.plot(x, test_loss, label='Test Loss')
# 添加标题和坐标轴标签
plt.title('Test Loss Curve')
plt.xlabel('Epoch')
plt.ylabel('Loss')
# 显示图例
plt.legend()
# 显示图形
plt.show()
```
在这个代码示例中,我们首先定义了测试集的 Loss 值,并将其存储在列表 `test_loss` 中。然后,我们使用 `range(len(test_loss))` 定义了 x 轴的取值范围,用于绘制折线图。接下来,我们使用 `plt.plot(x, test_loss, label='Test Loss')` 绘制了测试集上 Loss 的折线图,并为其添加了标签。然后,我们使用 `plt.title`、`plt.xlabel` 和 `plt.ylabel` 分别为图形添加标题和坐标轴标签。最后,我们使用 `plt.legend()` 显示图例,并使用 `plt.show()` 显示图形。运行这个代码示例,就可以得到一个带有测试集上 Loss 折线图的 Matplotlib 图形。