首页 > 解决方案 > 如何使 sklearn 完成的 confsuion 矩阵图更大,以一种正常方式包装所有数字而不像我的结果?

问题描述

如何使 sklearn 完成的 confsuion 矩阵图更大,以一种以正常方式包装所有数字的方式,而不像我的结果。Hier 是我的代码:




         import sklearn
         import numpy as np
         from mlxtend.plotting import plot_confusion_matrix
         import matplotlib.pyplot as plt

         y_pred = model.predict(X_test)
         Y_pred_classes = np.argmax(y_pred,axis=1) 
         Y_true = np.argmax(Y_test,axis=1)
         confusion_mtx = sklearn.metrics.confusion_matrix(Y_true, Y_pred_classes) 
         plot_confusion_matrix(confusion_mtx, class_names = allCategories() )

该图不适合矩阵

标签: pythonpandasscikit-learnseaborn

解决方案


推荐阅读