zoukankan      html  css  js  c++  java
  • 把tensorboard的多条训练损失曲线画在一张图上

    在做实验的时候,通常需要对比分析不同模型的训练损失情况,tensorboard画出来的通常是不同的损失在不同的图上,那如果我们像把不同模型的训练损失画在同一张图进行对比分析,应该怎么处理呢?

    可以通过matplotlib模块进行操作。

    一、下载Tensorboard中的loss曲线的数据(下载方式如图所示)

    二、用matplotlib模块画图

    import csv
    import matplotlib.pyplot as plt
    import pandas as pd
    
    df1 = pd.read_csv('/home/azhihong/JL-DCF-pytorch/log/run16-07/07-16-training loss.csv')  # csv文件所在路径
    step1 = df1['Step'].values.tolist()
    loss1 = df1['Value'].values.tolist()
    
    df2 = pd.read_csv('/home/azhihong/JL-DCF-pytorch/log/run17-08/08-17-training loss.csv')
    step2 = df2['Step'].values.tolist()
    loss2 = df2['Value'].values.tolist()
    
    plt.plot(step1, loss1, label='JL_DCF')
    plt.plot(step2, loss2, label='Ours')
    plt.legend(fontsize=16)  # 图注的大小
    plt.show()

    结果入下图所示

    完结!

     

  • 相关阅读:
    vue中路由跳转传递参数
    父组件向子孙组件传递数据provide/inject
    微信、QQ等内置浏览器定位失败
    Java ArrayList类
    java 生成 [1, n] 之间的随机数
    Java 构造方法
    Java this关键字
    Java private关键字及作用
    Java 随笔
    Java 内存划分
  • 原文地址:https://www.cnblogs.com/elitphil/p/15158753.html
Copyright © 2011-2022 走看看