记一次CNN模型训练遇到的问题（初训练）-526互联

csv文件读写

读：

filename = "D:\\桌面文件\\大三上\\大三下\\人工智能\\练习赛数据\\test_data.csv"
df = pd.read_csv(filename)
print(df.info())#简要摘要 information
print(df.head())#输出数据前n行，默认为5
print(df.shape[0])

写：（注意这个方式是一行一行写入，newline=''是为了消除csv文件里的空行）

import csv #调用数据保存文件
import pandas as pd #用于数据输出
with open("D:\\桌面文件\\大三上\\大三下\\人工智能\\练习赛数据\\ans.csv","w",newline='') as csvfile: 
    writer = csv.writer(csvfile)
    writer.writerow(["id","label"])
    t=1
    for i in Ans:
        writer.writerow([t,int(i)])
        t+=1

关于这个tensor转numpy的格式

y.cpu().numpy() ---> y.cpu().detach().numpy()