如何将Sklearn的分类报告输出成csv文件？

Question

有谁知道无论如何都可以将分类报告输出为文本文件或 CSV 文件？

python 中的这一行 print(metrics.classification_report(y_test, y_pred)) 给了我分类报告。我希望这份报告采用 csv 格式。

我尝试复制和粘贴，但这些列会混在一起！任何帮助表示赞赏！

原文由 user8034918 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

该函数有一个参数可以解决这个确切的问题。

 import pandas as pd
from sklearn.metrics import classification_report

report_dict = classification_report(y_true, y_pred, output_dict=True)
pd.DataFrame(report_dict)

将字典转换为数据框后，您可以将其写入 csv，轻松绘制它，对其进行操作或其他任何操作。

原文由 Rabeez Riaz 发布，翻译遵循 CC BY-SA 4.0 许可协议