python使用sklearn的ConfusionMatrixDisplay来可视化混淆矩阵

Posted 2021-10-08 Data+Science+Insight

python使用sklearn的ConfusionMatrixDisplay来可视化混淆矩阵

目录

python使用sklearn的ConfusionMatrixDisplay来可视化混淆矩阵

#模型构建

#ConfusionMatrixDisplay来可视化混淆矩阵

---

#模型构建

print(__doc__)
from sklearn.datasets import fetch_openml
from sklearn.preprocessing import StandardScaler
from sklearn.pipeline import make_pipeline
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split

X, y = fetch_openml(data_id=1464, return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, stratify=y)

clf = make_pipeline(StandardScaler(), LogisticRegression(random_state=0))
clf.fit(X_train, y_train)

#ConfusionMatrixDisplay来可视化混淆矩阵

from sklearn.metrics import confusion_matrix
from sklearn.metrics import ConfusionMatrixDisplay

y_pred = clf.predict(X_test)
cm = confusion_matrix(y_test, y_pred)

cm_display = ConfusionMatrixDisplay(cm).plot()

参考：ConfusionMatrixDisplay

参考：混淆矩阵