From 242eb958ddc4e08759349687b90f1555f0f4e23a Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 26 Dec 2019 17:43:27 +0100 Subject: cat --- inversion/live.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inversion/live.py b/inversion/live.py index 97078ae..ffac0af 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -244,7 +244,7 @@ class Interpolator: def set_category(self, category): print("Set category: {}".format(category)) - categories = " ".split(category) + categories = category.split(" ") label = np.zeros((BATCH_SIZE, N_CLASS,)) for category in categories: index = int(category) -- cgit v1.2.3-70-g09d2