diff options
Diffstat (limited to 'inversion/live.py')
| -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) |