Spaces:
Running
on
Zero
Running
on
Zero
increase th cluster sorting
Browse files
app.py
CHANGED
@@ -527,7 +527,7 @@ def make_cluster_plot(eigvecs, images, h=64, w=64, progess_start=0.6, advanced=F
|
|
527 |
# tensor = tensor[torch.randperm(tensor.shape[0])[:max_size]]
|
528 |
# return tensor.quantile(p)
|
529 |
# top_p = top_percentile(heatmap, p=0.5)
|
530 |
-
top_p = 0.
|
531 |
|
532 |
heatmap = heatmap.reshape(-1, h, w)
|
533 |
mask = (heatmap > top_p).float()
|
|
|
527 |
# tensor = tensor[torch.randperm(tensor.shape[0])[:max_size]]
|
528 |
# return tensor.quantile(p)
|
529 |
# top_p = top_percentile(heatmap, p=0.5)
|
530 |
+
top_p = 0.9
|
531 |
|
532 |
heatmap = heatmap.reshape(-1, h, w)
|
533 |
mask = (heatmap > top_p).float()
|