评价分类结果（1）--混淆矩阵、精准率、召回率、F1 Score

1. 导入数据

from sklearn import datasets

digits = datasets.load_digits()
X = digits.data
y = digits.target.copy()

y[digits.target==9] = 1
y[digits.target!=9] = 0

2 .数据分割

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=666)

3. 逻辑回归

from sklearn.linear_model import LogisticRegression

log_reg = LogisticRegression()
log_reg.fit(X_train,y_train)

保存预测结果

y_predict = log_reg.predict(X_test)

混淆矩阵

from sklearn.metrics import confusion_matrix

confusion_matrix(y_test,y_predict)

精确率

from sklearn.metrics import precision_score

precision_score(y_test,y_predict)

召回率

from sklearn.metrics import recall_score

recall_score(y_test,y_predict)

F1 score

from sklearn.metrics import f1_score

f1_score(y_test,y_predict)