diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-19 21:24:23 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-19 21:24:23 +0100 |
| commit | de2c68fa48524c37a59e576e0352788941ed9c36 (patch) | |
| tree | 5b50ecbebba3d8b220cb73a3d8ee41f5871c65e5 /inversion | |
| parent | 9a3c95e2274d770173c508e4da9fbdcd07f3fa5f (diff) | |
fix class
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 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])) |