Python读取log文件绘制Loss曲线

鲁棒最小二乘支持向量机

已于 2022-04-13 22:05:14 修改

阅读量2.7k

点赞数 4

分类专栏：计算机视觉笔记一起学Python 文章标签： python 深度学习

于 2022-04-13 21:45:35 首次发布

本文链接：https://blog.csdn.net/qq_42078934/article/details/124158612

版权

笔记同时被 3 个专栏收录

223 篇文章 167 订阅

订阅专栏

一起学Python

16 篇文章 0 订阅

订阅专栏

计算机视觉

13 篇文章 0 订阅

订阅专栏

训练model时将新鲜保存至log：

在这里插入图片描述

log文件存放数据格式如下：

完整程序如下：

import matplotlib.pyplot as plot
def getLoss(logFile):
    f = open(logFile, "r", encoding='utf-8')
    line = f.readline()  # 以行的形式进行读取文件
    lossArr = []
    loss2Arr = []
    loss3Arr = []
    loss4Arr = []
    loss5Arr = []
    while line:
        if 'loss=' in line:
            lossArr.append(float(line[-77:-68]))
            loss2Arr.append(float(line[-60:-51]))
            loss3Arr.append(float(line[-43:-34]))
            loss4Arr.append(float(line[-26:-17]))
            loss5Arr.append(float(line[-9:]))
        line = f.readline()
    f.close()
    return lossArr, loss2Arr, loss3Arr, loss4Arr, loss5Arr
def drawLine(arr, xName, yName, title, graduate):
    # 横坐标 采用列表表达式
    x = [x+1 for x in range(len(arr))]
    # 纵坐标
    y = arr
    # 生成折线图：函数polt
    plot.plot(x, y)
    # 设置横坐标说明
    plot.xlabel(xName)
    # 设置纵坐标说明
    plot.ylabel(yName)
    # 添加标题
    plot.title(title)
    # 设置纵坐标刻度
    plot.yticks(graduate)
    # 显示网格
    plot.grid(True)
    # 显示图表
    plot.show()
if __name__ == '__main__':
    loss, loss2, loss3, loss4, loss5 = getLoss(r"train_scwssod.log")
    print(loss)
    print(loss2)
    print(loss3)
    print(loss4)
    print(loss5)
    graduate = []
    deGraduate = 5
    # 计算y的刻度值
    for i in range(len(loss)):
        if i * deGraduate < max(loss) + deGraduate:
            graduate.append(i * deGraduate)
    print(graduate)
    drawLine(loss, "Iterated", "Loss", "Loss function curve", graduate)
    drawLine(loss2, "Iterated", "Loss", "Loss function curve", graduate)
    drawLine(loss3, "Iterated", "Loss", "Loss function curve", graduate)
    drawLine(loss4, "Iterated", "Loss", "Loss function curve", graduate)
    drawLine(loss5, "Iterated", "Loss", "Loss function curve", graduate)