Spaces:
Sleeping
Sleeping
aswin-raghavan
commited on
Commit
·
fc18f1c
1
Parent(s):
e23ed18
bugfix unique
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ def quantize_embeds(embeds):
|
|
91 |
def update_exemplars(df, rng, exemplars, lut):
|
92 |
embeds = df[['image_embed']].to_numpy()
|
93 |
labels = df[['label']].to_numpy()
|
94 |
-
assert np.all(
|
95 |
labels_zero_idx = (labels == 0).nonzero()
|
96 |
labels_one_idx = (labels == 1).nonzero()
|
97 |
print(labels_zero_idx.shape, " zeros and ", labels_one_idx.shape, " ones")
|
|
|
91 |
def update_exemplars(df, rng, exemplars, lut):
|
92 |
embeds = df[['image_embed']].to_numpy()
|
93 |
labels = df[['label']].to_numpy()
|
94 |
+
assert np.all(np.unique(labels) == [0, 1])
|
95 |
labels_zero_idx = (labels == 0).nonzero()
|
96 |
labels_one_idx = (labels == 1).nonzero()
|
97 |
print(labels_zero_idx.shape, " zeros and ", labels_one_idx.shape, " ones")
|