이번에는 혼동 행렬을 이용한 결과를 살펴보겠습니다.
코드 3-15 혼동 행렬을 이용한 성능 측정
from sklearn.metrics import confusion_matrix
pd.DataFrame(
confusion_matrix(y_test, y_predict),
columns=['Predicted Not Survival', 'Predicted Survival'],
index=['True Not Survival', 'True Survival']
)
다음 그림은 혼동 행렬에 대한 출력 결과입니다.
▲ 그림 3-16 결정 트리 코드 실행 결과
결과가 제대로 나왔는지 확인하려면 혼동 행렬의 개념을 이해해야 합니다.