Allob commited on
Commit
be07201
·
1 Parent(s): eabf510

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def load_words_dataset():
22
  def prepare_umap():
23
  all_enc = model.encode(all_words)
24
  umap_3d = UMAP(n_components=3, init='random', random_state=0)
25
- proj_3d = umap_3d.fit_transform(all_enc)
26
  return umap_3d
27
 
28
 
 
22
  def prepare_umap():
23
  all_enc = model.encode(all_words)
24
  umap_3d = UMAP(n_components=3, init='random', random_state=0)
25
+ proj_3d = umap_3d.fit_transform(random.sample(all_enc.tolist(), k=2000))
26
  return umap_3d
27
 
28