python怎么绘制RMSE图
时间: 2023-10-11 14:13:35 浏览: 211
其可以直接绘制vaspkit生成的能带图,直接调用即可,可以批量生成能带图,Python源码.zip
5星 · 资源好评率100%
首先,你需要安装 matplotlib 库来进行绘图。你可以使用以下命令来安装:
```
pip install matplotlib
```
然后,你可以使用以下代码来绘制 RMSE 图:
```python
import matplotlib.pyplot as plt
# RMSE 数据
rmse = [1.2, 1.1, 1.0, 0.9, 0.8, 0.7, 0.6, 0.5, 0.4, 0.3]
# 对应的 epoch 数
epoch = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
# 绘制 RMSE 图
plt.plot(epoch, rmse)
# 添加图标题
plt.title('RMSE vs Epoch')
# 添加 x 轴标签
plt.xlabel('Epoch')
# 添加 y 轴标签
plt.ylabel('RMSE')
# 显示图像
plt.show()
```
这段代码将绘制一个 RMSE vs Epoch 的图表,其中 RMSE 数据在 y 轴,对应的 epoch 数在 x 轴。你可以根据你的实际情况修改 rmse 和 epoch 数组来绘制你需要的图表。
阅读全文