基于matplotlib绘制test loss折线图定义x轴的取值范围
时间: 2024-04-06 15:32:24 浏览: 83
要定义 x 轴的取值范围,你可以使用 `plt.xlim()` 函数。这个函数可以接受两个参数,分别表示 x 轴的最小值和最大值。例如,如果你想要将 x 轴的取值范围限制在 0 到 30 之间,你可以使用以下代码:
```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))
# 绘制折线图,并限制 x 轴的取值范围
plt.plot(x, test_loss, label='Test Loss')
plt.xlim(0, 30)
# 添加标题和坐标轴标签
plt.title('Test Loss Curve')
plt.xlabel('Epoch')
plt.ylabel('Loss')
# 显示图例
plt.legend()
# 显示图形
plt.show()
```
在这个示例中,我们使用 `plt.xlim(0, 30)` 将 x 轴的取值范围限制在 0 到 30 之间。这样,我们就可以控制 x 轴的范围,使图形更易读。
阅读全文