Python用log文件绘制损失、准确率曲线

一、导入包

from matplotlib import rcParams
import matplotlib.pyplot as plt
import re

二、读取文件

##显示中文
rcParams['font.family'] = 'sans-serif'
rcParams['font.sans-serif'] = 'SimSun,Times New Roman'

##读取log文件
logFile = r'...\log-20210818-090024.log' 
text = ''
file = open(logFile)
for line in file:
    text += line
file.close()

三、正则表达式匹配训练集、验证集每个批次的数据

all_list = re.findall('step - loss: .*[0-9]',text)  

结果

正则表达式解释：’step - loss: .*[0-9]‘
因为log文件中每个批次的训练、验证数据以’step - loss‘开头，数字结尾

四、从包含训练、验证数据的大列表中分割训练集损失、训练集准确率、验证集损失、验证集准确率

train_loss =