Python – 深度学习训练过程使用matplotlib.pyplot实时动态显示loss和acc曲线
1 深度学习训练过程中实时动态显示loss和acc曲线
可参考代码如下:
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
if __name__ == '__main__':
train_loss = 5
val_loss = 5
train_acc = 0.0
val_acc = 0.0
x = []
train_loss_list = []
val_loss_list = []
train_acc_list = []
val_acc_list = []
for epoch in range(200):
# 生成数据,此处应根据实际训练过程获取训练集loss和acc
# 以及验证集loss和acc
train_loss -= epoch * 0.1
val_loss -= epoch * 0.11
train_acc += epoch*0.01
val_acc += epoch*0.011
x.append(epoch)
train_loss_list.append(train_loss)
val_loss_list.append(val_loss)
train_acc_list.append(train_acc)
val_acc_list.append(val_acc)
plt.figure(figsize=(3, 6), dpi=100)
# 创建两行一列的图,并指定当前使用第一个图
plt.subplot(2, 1, 1)
try:
train_loss_lines.remove(train_loss_lines[0]) # 移除上一步曲线
val_loss_lines.remove(val_loss_lines[0])
except Exception:
pass
train_loss_lines = plt.plot(x, train_loss_list, 'r', lw=1) # lw为曲线宽度
val_loss_lines = plt.plot(x, val_loss_list, 'b', lw=1)
plt.title("loss")
plt.xlabel("epoch")
plt.ylabel("loss")
plt.legend(["train_loss",
"val_loss"])
# # 创建两行一列的图,并指定当前使用第二个图
plt.subplot(2, 1, 2)
try:
train_acc_lines.remove(train_acc_lines[0]) # 移除上一步曲线
val_acc_lines.remove(val_acc_lines[0])
except Exception:
pass
train_acc_lines = plt.plot(x, train_acc_list, 'r', lw=1) # lw为曲线宽度
val_acc_lines = plt.plot(x, val_acc_list, 'b', lw=1)
plt.title("acc")
plt.xlabel("epoch")
plt.ylabel("acc")
plt.legend(["train_acc",
"val_acc"])
plt.show()
plt.pause(0.1) # 图片停留0.1s
2 结果
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Python – 深度学习训练过程使用matplotlib.pyplot实时动态显示loss和acc曲线
原文链接:https://www.stubbornhuang.com/1445/
发布于:2021年07月19日 13:54:53
修改于:2023年06月26日 21:28:34
当前分类随机文章推荐
- Python - 读取视频为numpy数组以及将numpy数组转换为视频 阅读2919次,点赞0次
- Python - 使用python-opencv裁剪原视频为与视频高同宽的视频 阅读1817次,点赞0次
- Python:UnicodeEncodeError: 'gbk' codec can't encode character '\xbb' in position 12305,以及中文乱码的解决方案 阅读3434次,点赞0次
- opencv-python - 读取视频,不改变视频分辨率修改视频帧率 阅读5417次,点赞2次
- Python - ModuleNotFoundError: No module named 'skimage' 阅读564次,点赞0次
- Python - 使用代码判断当前Python版本号 阅读658次,点赞0次
- Python3爬虫 - 下载反盗链图片的方式 阅读3051次,点赞1次
- Python - 使用letter box方法缩放图片,防止图片缩放时失真 阅读394次,点赞0次
- Python - opencv-python保存视频时出现Failed to load OpenH264 library: openh264-1.8.0-win64.dll错误 阅读1462次,点赞0次
- 简单粗暴:使用pycharm安装对应的Python版本第三方包 阅读3939次,点赞0次
全站随机文章推荐
- C++ - RAII机制 阅读987次,点赞0次
- 资源分享 - Python深度学习:基于PyTorch (吴茂贵著) 高清PDF下载 阅读5143次,点赞0次
- Pytorch - Pytoch结合Tensorboard实现数据可视化 阅读439次,点赞0次
- Sigmoid激活函数的快速替代函数以及相应的C++实现 阅读2222次,点赞0次
- 资源分享 - Ray Tracing Gems II - Next Generation Real-Time Rendering with DXR, Vulkan, and OptiX-Apress 英文高清PDF下载 阅读2333次,点赞0次
- Centos7 - 安装高版本的gcc/g++,多版本切换以及配置环境 阅读69次,点赞0次
- 资源分享 - Graphics Programming Methods 英文高清PDF下载 阅读2185次,点赞0次
- 资源分享 - Graphics Shaders - Theory and Practice (Second Edition) 英文高清PDF下载 阅读2660次,点赞0次
- WordPress - 用户修改密码/邮箱时禁止向管理员/用户发送通知邮件 阅读1141次,点赞0次
- 资源分享 - 实时相机处理技术,Real-Time Cameras中文版PDF下载 阅读2246次,点赞0次
评论
169