Sto usando scikit-learn per la classificazione di documenti di testo (da 22000) a 100 classi. Uso il metodo della matrice di confusione di scikit-learn per calcolare la matrice di confusione.Come posso tracciare una matrice di confusione?
model1 = LogisticRegression()
model1 = model1.fit(matrix, labels)
pred = model1.predict(test_matrix)
cm=metrics.confusion_matrix(test_labels,pred)
print(cm)
plt.imshow(cm, cmap='binary')
Ecco come il mio matrice di confusione si presenta come:
[[3962 325 0 ..., 0 0 0]
[ 250 2765 0 ..., 0 0 0]
[ 2 8 17 ..., 0 0 0]
...,
[ 1 6 0 ..., 5 0 0]
[ 1 1 0 ..., 0 0 0]
[ 9 0 0 ..., 0 0 9]]
Tuttavia, non ricevono una trama trasparente o leggibili. C'è un modo migliore per farlo?