我正在尝试confusion matrix使用数据集绘制Logistic Regression一个multi-class。
但问题是当我绘制confusion matrix它时它只绘制一个confusion matrix用于二进制分类的。
这是我正在绘制它的地方。
%matplotlib inline
import matplotlib.pyplot as plt
import pandas as pd
dataframe = pd.read_csv("WA_Fn-UseC_-HR-Employee-Attrition.csv")
from sklearn.linear_model import LogisticRegression
LRModel = LogisticRegression(C=100, max_iter=5500)
LRModel.fit(X_train, y_train)
predicted_values_ = LRModel.predict(X_test)
from sklearn.metrics import confusion_matrix
cm = confusion_matrix(y_test, predicted_values_)
misclassified = (y_test != predicted_values_).sum()
misclassified
import seaborn as sn
# plt.figure()
sn.heatmap(cm, annot=True)
plt.xlabel("Predicted")
plt.ylabel("Actual")
我得到这个矩阵,如下所示。
有人可以告诉我我在哪里做错了吗?
这是我在多类场景中使用逻辑回归的地方
