当前位置:   article > 正文

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

train_acc

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.代码(示例):

  1. import pandas as pd
  2. from datetime import datetime
  3. # 1.新建csv文件,用于后续记录time、epoch、loss、accuracy等结果。
  4. df = pd.DataFrame(columns=['time', 'epoch', 'train_Loss', 'train_acc']) # 列名
  5. df.to_csv("./train_acc.csv", index=False) # 路径可以根据需要更改
  6. # 2.获取数据,将结果写入csv文件中。(注:这里仅作举例说明,所以train_Loss、train_acc都是随便写的固定数字)
  7. for i in range(10): #假设迭代10次
  8. # 1)获取数据
  9. time = "%s"%datetime.now() #获取当前时间
  10. epoch = i
  11. train_Loss = 0.23333
  12. train_acc = 0.82222
  13. # 2)将数据保存在一维列表,然后转化为DataFrame
  14. list = [time, epoch , train_Loss, train_acc]
  15. # 由于DataFrame是Pandas库中的一种数据结构,它类似excel,是一种二维表,所以需要将list以二维列表的形式转化为DataFrame
  16. data = pd.DataFrame([list])
  17. # 3)将数据写入csv文件
  18. 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等数据保存到文件

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/129794
推荐阅读
相关标签
  

闽ICP备14008679号