亲宝软件园·资讯

展开

python classification_report输出csv python 怎样把classification_report输出到csv文件

农民小飞侠 人气:0
想了解python 怎样把classification_report输出到csv文件的相关内容吗,农民小飞侠在本文为您仔细讲解python classification_report输出csv的相关知识和一些Code实例,欢迎阅读和指正,我们先划重点: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)

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

补充:sklearn classification_report 输出说明

svm-rbf 0.606
precision recall f1-score support
0.0 0.56 0.39 0.46 431
1.0 0.62 0.77 0.69 569
avg / total 0.60 0.61 0.59 1000

最后一行是用support 加权平均算出来的,如0.59 = (431*0.46+569*0.69)/ 1000

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。

加载全部内容

相关教程
猜你喜欢
用户评论