Merge pull request #51 from Huang-Jack/patch-4

add confusion_matrix output and print classification report
2018-08-22 13:48:04 +08:00
parent ba982d4ec1 aed67ef41e
commit 62e0010893
1 changed files with 3 additions and 0 deletions
--- a/11_k-NN.py
+++ b/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))