Tweaks

Author: GraemeMalcolm

03 - Classification.ipynb

@@ -431,6 +431,9 @@
         "print(classification_report(y_test, predictions))\n",
         "print (\"Confusion Matrix:\\n\",cm)\n",
         "\n",
+        "auc = roc_auc_score(y_test,y_scores[:,1])\n",
+        "print('\\nAUC: ' + str(auc))\n",
+        "\n",
         "# calculate ROC curve\n",
         "y_scores = model.predict_proba(X_test)\n",
         "fpr, tpr, thresholds = roc_curve(y_test, y_scores[:,1])\n",
@@ -444,7 +447,8 @@
         "plt.xlabel('False Positive Rate')\n",
         "plt.ylabel('True Positive Rate')\n",
         "plt.title('ROC Curve')\n",
-        "plt.show()"
+        "plt.show()\n",
+        "\n"
       ]
     },
     {
@@ -495,6 +499,8 @@
         "print('Accuracy: ', accuracy_score(y_test, predictions))\n",
         "print(classification_report(y_test, predictions))\n",
         "print (\"Confusion Matrix:\\n\",cm)\n",
+        "auc = roc_auc_score(y_test,y_scores[:,1])\n",
+        "print('\\nAUC: ' + str(auc))\n",
         "\n",
         "# calculate ROC curve\n",
         "y_scores = model.predict_proba(X_test)\n",