Python: トレーニング損失、train_acc、およびその他のデータを csv ファイルに書き込みます

1. 目標:

トレーニング中のエポック、トレーニング後に得られた損失、train_acc、およびその他の結果を csv ファイルに書き込みます。

2. 実装手順:

1. 時間、エポック、損失、精度、その他の結果を後で記録するために、新しい csv ファイルを作成します。

2. データを取得し、結果を csv ファイルに書き込みます。 (注: これは単なる例なので、train_Loss と train_acc は適当に書いた固定の数字です)。

1. データを取得します。

2. データを 1 次元リストに保存し、DataFrame に変換します。

3. 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 トレーニングとテストの損失、精度、その他のデータをファイルに保存する