From 9a3c95e2274d770173c508e4da9fbdcd07f3fa5f Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Thu, 19 Dec 2019 21:20:23 +0100
Subject: fix class

---
 inversion/live.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'inversion/live.py')

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]))
-- 
cgit v1.2.3-70-g09d2