Add a comment about sklearn.metrics.plot_confusion_matrix, fixes #142

2020-07-05 18:35:06 +12:00 · 2020-07-05 18:35:06 +12:00 · 4a48d04a1a
parent cdfad2d003
commit 4a48d04a1a
1 changed files with 1 additions and 0 deletions
--- a/03_classification.ipynb
+++ b/03_classification.ipynb
@ -852,6 +852,7 @@
   "metadata": {},
   "outputs": [],
   "source": [
+    "# since sklearn 0.22, you can use sklearn.metrics.plot_confusion_matrix()\n",
    "def plot_confusion_matrix(matrix):\n",
    "    \"\"\"If you prefer color and a colorbar\"\"\"\n",
    "    fig = plt.figure(figsize=(8,8))\n",