python:将训练得到的loss、train_acc等数据写入csv文件

本文介绍了如何使用Python的pandas库将训练模型时的epoch数、loss和accuracy等信息保存到CSV文件中。通过创建DataFrame,每次迭代后追加数据至文件,以便于后期分析和可视化。
摘要由CSDN通过智能技术生成

1.目标:

将训练时的epoch,以及训练后得到的loss、train_acc等结果写入csv文件。

2.实现步骤:

  1. 新建csv文件,用于后续记录time、epoch、loss、accuracy等结果。

  1. 获取数据,将结果写入csv文件中。(注:这里仅作举例说明,所以train_Loss、train_acc都是随便写的固定数字)。

  1. 获取数据。

  1. 将数据保存在一维列表,然后转化为DataFrame。

  1. 将DataFrame数据写入csv文件。

3.代码(示例):

import pandas as pd
from datetime import datetime

# 1.新建csv文件,用于后续记录time、epoch、loss、accuracy等结果。
df = pd.DataFrame(columns=['time', 'epoch', 'train_Loss', 'train_acc'])  # 列名
df.to_csv("./train_acc.csv", index=False)  # 路径可以根据需要更改

# 2.获取数据,将结果写入csv文件中。(注:这里仅作举例说明,所以train_Loss、train_acc都是随便写的固定数字)
for i in range(10): #假设迭代10次
    # 1)获取数据
    time = "%s"%datetime.now() #获取当前时间
    epoch = i
    train_Loss = 0.23333
    train_acc = 0.82222

    # 2)将数据保存在一维列表,然后转化为DataFrame
    list = [time, epoch , train_Loss, train_acc]
    # 由于DataFrame是Pandas库中的一种数据结构,它类似excel,是一种二维表,所以需要将list以二维列表的形式转化为DataFrame
    data = pd.DataFrame([list])
    # 3)将数据写入csv文件
    data.to_csv('./train_acc.csv', mode='a', header=False, index=False)  # mode设为a,就可以向csv文件追加数据了

4.结果:

参考:

https://blog.csdn.net/weixin_43760844/article/details/113245871 将训练和测试的loss、accuracy等数据保存到文件

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

weixin_39450145

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值