fix class

1 files changed, 1 insertions, 1 deletions

diff --git a/inversion/live.py b/inversion/live.py
index be03422..6894b47 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -67,7 +67,7 @@ def clamp(n, a=0, b=1):
 def label_sampler(num_classes=1, shape=(BATCH_SIZE, N_CLASS,)):
   label = np.zeros(shape)
   for i in range(shape[0]):
-    for _ in range(random.randint(1, shape[1])):
+    for _ in range(num_classes):
       j = random.randint(0, shape[1]-1)
       label[i, j] = random.random()
       print("class: {} {}".format(j, label[i, j]))