相关疑难解决方法(0)

sklearn情节混淆矩阵与标签

我想绘制一个混淆矩阵来显示分类器的性能,但它只显示标签的数量,而不是标签本身:

from sklearn.metrics import confusion_matrix
import pylab as pl
y_test=['business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business', 'business']

pred=array(['health', 'business', 'business', 'business', 'business',
       'business', 'health', 'health', 'business', 'business', 'business',
       'business', 'business', 'business', 'business', 'business',
       'health', 'health', 'business', 'health'], 
      dtype='|S8')

cm = confusion_matrix(y_test, pred)
pl.matshow(cm)
pl.title('Confusion matrix of the classifier')
pl.colorbar()
pl.show()
Run Code Online (Sandbox Code Playgroud)

如何将标签(health,business..etc)添加到混淆矩阵中?

python matplotlib scikit-learn

51
推荐指数
8
解决办法
10万
查看次数

标签 统计

matplotlib ×1

python ×1

scikit-learn ×1