From 6ea02a224d445251701100cd95740fe29fa7e564 Mon Sep 17 00:00:00 2001
From: Pierluigi Ferrari <pierluigi.ferrari@gmx.com>
Date: Thu, 26 Apr 2018 15:01:45 +0200
Subject: [PATCH] chore: Fix precision-recall plots for the cases of 1 and 2
 classes

---
 ssd300_evaluation.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ssd300_evaluation.ipynb b/ssd300_evaluation.ipynb
index c25554d1..7aebee18 100644
--- a/ssd300_evaluation.ipynb
+++ b/ssd300_evaluation.ipynb
@@ -401,7 +401,7 @@
     }
    ],
    "source": [
-    "m = (n_classes + 1) // 2\n",
+    "m = max((n_classes + 1) // 2, 2)\n",
     "n = 2\n",
     "\n",
     "fig, cells = plt.subplots(m, n, figsize=(n*8,m*8))\n",