huzey commited on
Commit
4d75059
1 Parent(s): b50d4dc

increase th cluster sorting

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.8
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()