OVR, OVO, Multiclass AUC

Author: GraemeMalcolm

03 - Classification.ipynb

@@ -1075,7 +1075,7 @@
     "print(\"Overall Accuracy:\",accuracy_score(y_penguin_test, penguin_predictions))\n",
     "print(\"Overall Precision:\",precision_score(y_penguin_test, penguin_predictions, average='macro'))\n",
     "print(\"Overall Recall:\",recall_score(y_penguin_test, penguin_predictions, average='macro'))\n",
-    "print('Average AUC:', auc)\n",
+    "print('Average AUC:', roc_auc_score(y_penguin_test,penguin_prob, multi_class='ovr'))\n",
     "\n",
     "# Confusion matrix\n",
     "plt.imshow(mcm, interpolation=\"nearest\", cmap=plt.cm.Blues)\n",