diff options
Diffstat (limited to 'inversion')
| -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 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])) |