20200813 -
引言
- 打印分类报告
问题
1. 打印分类报告
在分类过程结束之后,需要反馈分类效果,使用的函数是classification_report
[1]。函数的全部分信息如下
sklearn.metrics.classification_report(y_true, y_pred, *, labels=None, target_names=None, sample_weight=None, digits=2, output_dict=False, zero_division='warn')
平时使用的过程中有两个需求,1)打印出来的时候,需要分类的字符串标签,这个部分由target_names传入;2)打印出来的数值只有小数点两位的信息,一般还需要更多的信息,该部分由digits参数控制,传入digits=4即可。
在使用过程中,还遇到一个问题,有时分类可能某个类别全部全军覆没了,此时target_names传入的类别就不能匹配了,就会报错。要解决这个问题,只需要传递labels一个列表即可,就是把所有的要标记的索引都传递进去。
参考
[1]sklearn.metrics.classification_report