Python读取log文件绘制Loss曲线

谁借莪１个温暖的怀抱￠ 2023-09-29 10:08 50阅读 0赞

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

![在这里插入图片描述][watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBA6bKB5qOS5pyA5bCP5LqM5LmY5pSv5oyB5ZCR6YeP5py6_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center]

## 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)

**本文希望对大家有帮助，当然上文若有不妥之处，欢迎指正。**

**分享决定高度，学习拉开差距**

[watermark_type_d3F5LXplbmhlaQ_shadow_50_text_Q1NETiBA6bKB5qOS5pyA5bCP5LqM5LmY5pSv5oyB5ZCR6YeP5py6_size_20_color_FFFFFF_t_70_g_se_x_16_pic_center]: https://img-blog.csdnimg.cn/723bcd050c684fc3bac6431ed0219126.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA6bKB5qOS5pyA5bCP5LqM5LmY5pSv5oyB5ZCR6YeP5py6,size_20,color_FFFFFF,t_70,g_se,x_16#pic_center