如何通过python画loss曲线的方法

 更新时间:2019年06月26日 14:27:30   作者:Qing_414  
这篇文章主要介绍了如何通过python画loss曲线的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

1. 首先导入一些python画图的包,读取txt文件,假设我现在有两个模型训练结果的records.txt文件

import numpy as np
import matplotlib.pyplot as plt
import pylab as pl
from mpl_toolkits.axes_grid1.inset_locator import inset_axes
data1_loss =np.loadtxt("valid_RCSCA_records.txt") 
data2_loss = np.loadtxt("valid_SCRCA_records.txt") 

2. 我自己的数据第一列是训练步数,第二列的loss,所以取出相应列的数据作为绘图的x和y

x = data1_loss[:,0]
y = data1_loss[:,1]
x1 = data2_loss[:,0]
y1 = data2_loss[:,1]

3. 先创建一幅图,再在这幅图上添加一个小图,小图用来显示部分放大的曲线

fig = plt.figure(figsize = (7,5))    #figsize是图片的大小`
ax1 = fig.add_subplot(1, 1, 1) # ax1是子图的名字`

4. 先画出整体的loss曲线

pl.plot(x,y,'g-',label=u'Dense_Unet(block layer=5)')`
# ‘'g‘'代表“green”,表示画出的曲线是绿色,“-”代表画的曲线是实线,可自行选择,label代表的是图例的名称,一般要在名称前面加一个u,如果名称是中文,会显示不出来,目前还不知道怎么解决。
p2 = pl.plot(x1, y1,'r-', label = u'RCSCA_Net')
pl.legend()
#显示图例
p3 = pl.plot(x2,y2, 'b-', label = u'SCRCA_Net')
pl.legend()
pl.xlabel(u'iters')
pl.ylabel(u'loss')
plt.title('Compare loss for different models in training')

画出曲线如图:


5. 显示放大的部分曲线

# plot the box
tx0 = 0
tx1 = 10000
#设置想放大区域的横坐标范围
ty0 = 0.000
ty1 = 0.12
#设置想放大区域的纵坐标范围
sx = [tx0,tx1,tx1,tx0,tx0]
sy = [ty0,ty0,ty1,ty1,ty0]
pl.plot(sx,sy,"purple")
axins = inset_axes(ax1, width=1.5, height=1.5, loc='right')
#loc是设置小图的放置位置,可以有"lower left,lower right,upper right,upper left,upper #,center,center left,right,center right,lower center,center"
axins.plot(x1,y1 , color='red', ls='-')
axins.plot(x2,y2 , color='blue', ls='-')
axins.axis([0,20000,0.000,0.12])
plt.savefig("train_results_loss.png")
pl.show
#pl.show()也可以

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。

相关文章

  • Spring实战之使用util:命名空间简化配置操作示例

    Spring实战之使用util:命名空间简化配置操作示例

    这篇文章主要介绍了Spring实战之使用util:命名空间简化配置操作,结合实例形式分析了Spring使用util:命名空间简化配置操作的具体步骤与相关操作注意事项,需要的朋友可以参考下
    2019-12-12
  • Python自动化办公之合并多个Excel

    Python自动化办公之合并多个Excel

    在日常的办公自动化工作中,尤其是处理大量数据时,合并多个 Excel 表格是一个常见且繁琐的任务,下面小编就来为大家介绍一下如何使用Python轻松实现合并多个Excel吧
    2025-02-02
  • Python从临时邮箱获取验证码的操作代码

    Python从临时邮箱获取验证码的操作代码

    这篇文章主要介绍了Python从临时邮箱获取验证码的操作代码,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2023-08-08
  • 详解Python中range()与xrange()的区别

    详解Python中range()与xrange()的区别

    range() 和 xrange() 是两个函数,可用于在 Python的 for 循环中迭代一定次数。本文将通过示例详细说说二者的区别与使用,需要的可以参考一下
    2022-09-09
  • Python Pygame实战之五款童年经典游戏合集

    Python Pygame实战之五款童年经典游戏合集

    本文为大家总结了五款利用Python+Pygame实现的童年经典游戏:推箱子、滑雪、八分音符酱、保卫萝卜和飞机大战,快跟随小编一起学习一下
    2022-04-04
  • Python实现在某个数组中查找一个值的算法示例

    Python实现在某个数组中查找一个值的算法示例

    今天小编就为大家分享一篇Python实现在某个数组中查找一个值的算法示例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-06-06
  • django框架单表操作之增删改实例分析

    django框架单表操作之增删改实例分析

    这篇文章主要介绍了django框架单表操作之增删改,结合实例形式分析了Django框架前台数据操作提交与后台处理相关实现技巧,需要的朋友可以参考下
    2019-12-12
  • Python学习笔记之集合的概念和简单使用示例

    Python学习笔记之集合的概念和简单使用示例

    这篇文章主要介绍了Python学习笔记之集合的概念和简单使用,涉及Python集合的定义、查找、添加、删除等相关操作技巧与注意事项,需要的朋友可以参考下
    2019-08-08
  • 在cmd命令行里进入和退出Python程序的方法

    在cmd命令行里进入和退出Python程序的方法

    今天小编就为大家分享一篇在cmd命令行里进入和退出Python程序的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-05-05
  • 教你python 中如何取出colomap部分的颜色范围

    教你python 中如何取出colomap部分的颜色范围

    这篇文章主要介绍了python 中如何取出colomap部分的颜色范围,本文以以jet为例给大家提供一种方法,可以提取colormap色标中的一部分,取出我们满意的色标区域,感兴趣的朋友跟随小编一起看看吧
    2022-02-02

最新评论