将 DNN 输出写入 csv 文件

Question

我正在做我训练有素的深度神经网络的推理阶段。我的模型的输出是这样的：

tensor([19])
tensor([25])
tensor([18])

我想将原始输出保存在 CSV 文件中以进行一些统计分析。我试过这个

for data in (dataloader):
 z_eval = model_back(enhanced.float().to(device))
    torch.cuda.empty_cache()
    gc.collect()

    pred = [torch.max(z.detach().cpu(), dim=1)[1] for z in z_eval]
    torch.cuda.empty_cache()
    gc.collect()
    pred_test = pred[0]
    print(pred_test)
    #np.savetxt('output.csv', pred_test,delimiter= ',')
    prediction = pd.DataFrame(pred_test, columns=['predictions']).to_csv('prediction.csv')
    

问题是它只保存最后一个值而不是所有值，所以我怎样才能强制它保存所有值。

Answer 1

使用列表保存数据并在for之后保存到csv

pred_test =[]

for data in (dataloader):
  z_eval = model_back(enhanced.float().to(device))
  torch.cuda.empty_cache()
  gc.collect()
  pred = [torch.max(z.detach().cpu(), dim=1)[1] for z in z_eval]
  torch.cuda.empty_cache()
  gc.collect()
  pred_test += pred[0]
  print(pred_test)
  #np.savetxt('output.csv', pred_test,delimiter= ',')
prediction = pd.DataFrame(pred_test, columns=['predictions']).to_csv('prediction.csv')