python classification_report输出到csv文件

今天想把classification_report的统计结果输出到文件中,我这里分享一下一个简洁的方式:

我的pandas版本:

pandas                   1.0.3

代码:

from sklearn.metrics import classification_report
report = classification_report(y_test, y_pred, output_dict=True)
df = pd.DataFrame(report).transpose()
df.to_csv("result.csv", index= True)

是不是很简单,下面是我导出来的一个结果:

python classification_report输出到csv文件_第1张图片

参考文献

[1].scikit learn output metrics.classification_report into CSV/tab-delimited format. https://stackoverflow.com/questions/39662398/scikit-learn-output-metrics-classification-report-into-csv-tab-delimited-format

你可能感兴趣的:(scikit-learn)