실제 Nagative인 데이터 중 Positive로 잘못 예측한 비율
분류모델은 그 결과를 속할 비율(확률)을 반환한다
- 지금까지는 비율에서 threshold를 0.5라고 하고 0, 1로 결과를 반영했다. (이진분류)
- iris의 경우 가장 높은 확률값이 있는 클래스를 해당 값이라고 했다
실제 양성인 데이터를 음성이라고 판단하면 안되는 경우라면, Recall이 중요
- 이경우는 Threshold를 0.3 or 0.4로 선정해야 함. (ex. 암환자 판별)
실제 음성인 데이터를 양성이라고 판단하면 안되는 경우라면, Precision이 중요
- 이경우는 Threshold를 0.8 or 0.9로 선정해야 함. (ex. 스팸메일)
그러나 Recall과 Precision은 서로 영향을 주기 때문에 한 쪽을 극단적으로 높게 설정해서는 안된다
import pandas as pd
red_url = 'https://raw.githubusercontent.com/PinkWink/ML_tutorial/master/dataset/winequality-red.csv'
white_url = 'https://raw.githubusercontent.com/PinkWink/ML_tutorial/master/dataset/winequality-white.csv'
red_wine = pd.read_csv(red_url, sep= ';')
white_wine = pd.read_csv(white_url, sep= ';')
red_wine['color'] = 1.
white_wine['color'] = 0.
wine = pd.concat([red_wine, white_wine])
wine['taste'] = [1. if grade > 5 else 0. for grade in wine['quality']]
X = wine.drop(['taste', 'quality'], axis = 1)
y = wine['taste']
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import accuracy_score
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=13)
wine_tree = DecisionTreeClassifier(max_depth=2, random_state=13)
wine_tree.fit(X_train, y_train)
y_pred_tr = wine_tree.predict(X_train)
y_pred_test = wine_tree.predict(X_test)
print('train Acc : ',accuracy_score(y_train, y_pred_tr))
print('test Acc : ',accuracy_score(y_test, y_pred_test))
# train Acc : 0.7294593034442948
# test Acc : 0.7161538461538461
from sklearn.metrics import (accuracy_score, precision_score, recall_score, f1_score, roc_auc_score, roc_curve)
print('Accuracy :', accuracy_score(y_test, y_pred_test))
print('Recall :', recall_score(y_test, y_pred_test))
print('Precision :', precision_score(y_test, y_pred_test))
print('AUC Score :', roc_auc_score(y_test, y_pred_test))
print('F1 Score :', f1_score(y_test, y_pred_test))
# Accuracy : 0.7161538461538461
# Recall : 0.7314702308626975
# Precision : 0.8026666666666666
# AUC Score : 0.7105988470875331
# F1 Score : 0.7654164017800381
import matplotlib.pyplot as plt
%matplotlib inline
pred_proba = wine_tree.predict_proba(X_test)[:, 1]
fpr, tpr, threshold = roc_curve(y_test, pred_proba)
plt.figure(figsize=(10,8))
plt.plot([0,1],[0,1], ls='dashed')
plt.plot(fpr, tpr)
plt.grid()
plt.show()