PyTorch,把训练历史(损失函数和准确率)保存到csv文件中

一、首先生成一个CSV文件

import random
import pandas as pd
from datetime import datetime
#创建train_acc.csv和var_acc.csv文件，记录loss和accuracy
df = pd.DataFrame(columns=['time','step','train Loss','training accuracy'])#列名
df.to_csv("F:\\Documents\\train_acc.csv",index=False) #路径可以根据需要更改

二、将训练数据存入csv文件

#初始化train数据
t_loss = 0.4
t_acc = 0.3
for i in range(20):#假设迭代20次
    time = "%s"%datetime.now()#获取当前时间
    step = "Step[%d]"%i
    t_loss = t_loss - random.uniform(0.01,0.017)
    train_loss = "%f"%t_loss
    t_acc = t_acc + random.uniform(0.025,0.035)
    train_acc = "%g"%t_acc
#将数据保存在一维列表
    list = [time,step,train_loss,train_acc]
#由于DataFrame是Pandas库中的一种数据结构，它类似excel，是一种二维表，所以需要将list以二维列表的形式转化为DataFrame
    data = pd.DataFrame([list])
    data.to_csv('F:\\Documents\\train_acc.csv',mode='a',header=False,index=False)#mode设为a,就可以向csv文件追加数据了