diff options
| -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 6894b47..b5ac0c3 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(num_classes): + for _ in range(int(num_classes)): j = random.randint(0, shape[1]-1) label[i, j] = random.random() print("class: {} {}".format(j, label[i, j])) |