Сохранение DataFrame на каждой итерации

Question

Я пытаюсь сохранить некоторые значения из моей модели обучения как DataFrame, который работает нормально, но сохраняет данные только для первого l oop и останавливается. Я пытался вставить его в For l oop, но не помогло. Какие у меня есть способы сэкономить на всю тренировку?

 df=pd.DataFrame({'Image_Name':names_preds, 'Target':targets.detach(), 'Prediction':preds.detach()})
            print(df)
            df.to_csv(r'test5.csv', index=False)

Answer 1

Adding mode = 'a' решил проблему.

df=pd.DataFrame({'Image_Name':names_preds, 'Target':targets.detach(), 'Prediction':preds.detach()})
            print(df)
            df.to_csv(r'test5.csv',mode='a', index=False)