sklearn库使用问题汇总

20200813 -

引言

1. 打印分类报告

问题

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