今天想把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)
是不是很简单,下面是我导出来的一个结果:
参考文献
[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