def writeCsv(File,species):
row = [File,species]
out = open("data/test.csv", "a", newline="")
csv_writer = csv.writer(out, dialect="excel")
csv_writer.writerow(row)
#若想在csv文件第一行加标题 writeCsv("file","species")
#使用的时候直接调用就行,例如我写了一个代码中需要调用它,调用代码是
writeCsv(file, classes[pred.data.item()])
打开CSV文件,就是这样的