小编had*_*815的帖子

并排绘制多个混淆矩阵

我刚来这地方。这是我的第一个问题,希望得到专家的解答。我有 5 个分类器模型,我正在尝试绘制它们的混淆矩阵。

from sklearn.linear_model import LogisticRegression
from sklearn.svm import SVC
from sklearn.neighbors import KNeighborsClassifier
from sklearn.tree import DecisionTreeClassifier
from sklearn.ensemble import RandomForestClassifier
import collections

classifiers = {
    "Naive Bayes": GaussianNB(),
    "LogisiticRegression": LogisticRegression(),
    "KNearest": KNeighborsClassifier(),
    "Support Vector Classifier": SVC(),
    "DecisionTreeClassifier": DecisionTreeClassifier(),
}
Run Code Online (Sandbox Code Playgroud)

进而

from sklearn.metrics import confusion_matrix
for key, classifier in classifiers.items(): 
    y_pred = classifier.fit(X_train, y_train).predict(X_test)
    cf_matrix=confusion_matrix(y_test, y_pred)
    print(cf_matrix)
Run Code Online (Sandbox Code Playgroud)

这给了我 这

现在我尝试用下面的代码绘制它们,但图中没有显示数据。

fig, axn = plt.subplots(1,5, sharex=True, sharey=True)
cbar_ax = fig.add_axes([.91, .3, .03, .4])

for i, ax in enumerate(axn.flat):
    sns.heatmap(cf_matrix, ax=ax, …
Run Code Online (Sandbox Code Playgroud)

python plot matplotlib scikit-learn seaborn

3
推荐指数
1
解决办法
1万
查看次数

标签 统计

matplotlib ×1

plot ×1

python ×1

scikit-learn ×1

seaborn ×1