可以计算多标签分类的混淆矩阵的 Python 库

数据挖掘 Python 软件推荐 多标签分类
2021-09-30 09:20:28

我正在寻找一个可以计算多标签分类的混淆矩阵的 Python 库。


供参考:

4个回答

另请查看scikit-multilearn这是一个非常好的库,可以扩展 sklearn 以进行多标签学习。但是,我不确定混淆矩阵如何解决多标签问题......

这个人声称他已经解决了。

虽然这个问题很老,但我正在为新观众写这个答案。
scikit-learn 现在支持多标签分类的混淆矩阵。

https://scikit-learn.org/stable/modules/generated/sklearn.metrics.multilabel_confusion_matrix.html

有许多不同的参数可以通过比较真实标签和预测标签来评估您的方法的性能。我建议使用 PyCM模块,它可以提供多种适合多类分类的参数。

试试mlxtend是一个多类案例的例子。

在此处输入图像描述