点击查看代码
import csv
def save_class_accuracy_csv(class_accuracies):
# 将每个类别的准确度保存为 CSV 文件
with open('class_accuracies.csv', 'w', newline='') as csvfile:
fieldnames = ['Class', 'Accuracy']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
for i, acc in enumerate(class_accuracies):
writer.writerow({'Class': f'Class_{i}', 'Accuracy': acc})