From aed67ef41e263ddf451824e28e6a573da9cf9402 Mon Sep 17 00:00:00 2001 From: Jack Huang Date: Wed, 22 Aug 2018 13:41:14 +0800 Subject: [PATCH] add confusion_matrix output and print classification report @zhyongquan please involve me in the project, I'd like to contribute my effort to translate some chapters. --- Code/Day 11_k-NN.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Code/Day 11_k-NN.py b/Code/Day 11_k-NN.py index 6c258b2..b89e6b7 100644 --- a/Code/Day 11_k-NN.py +++ b/Code/Day 11_k-NN.py @@ -28,4 +28,7 @@ y_pred = classifier.predict(X_test) # Making the Confusion Matrix from sklearn.metrics import confusion_matrix +from sklearn.metrics import classification_report cm = confusion_matrix(y_test, y_pred) +print(cm) +print(classification_report(y_test, y_pred))