Merge pull request #48 from Huang-Jack/patch-1

enhance to print confusion_matrix and classification report
2018-08-21 14:40:21 +08:00
parent a7a0ce187c 408b05784e
commit 4956b84872
1 changed files with 3 additions and 0 deletions
--- a/6_Logistic_Regression.py
+++ b/6_Logistic_Regression.py
@ -28,5 +28,8 @@ 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 confusion_matrix
+print(classification_report(y_test, y_pred))   # print classification report