diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-26 17:43:27 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-26 17:43:27 +0100 |
| commit | 242eb958ddc4e08759349687b90f1555f0f4e23a (patch) | |
| tree | 93087b822786f5c4b19bf6577ced62773f9a4402 | |
| parent | 49eca5f936cea9900f878018e9c76ba4b752f447 (diff) | |
cat
| -rw-r--r-- | inversion/live.py | 2 |
1 files changed, 1 insertions, 1 deletions
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) |