de-Rodrigo commited on
Commit
42cb4fb
·
1 Parent(s): 1ac3193

Try Different N Neighbours UMAP

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -56,6 +56,7 @@ class RelativeScaler(BaseEstimator, TransformerMixin):
56
  N_COMPONENTS = 3
57
  TSNE_NEIGHBOURS = 15
58
  # WEIGHT_FACTOR = 0.05
 
59
 
60
 
61
  TOOLTIPS = """
@@ -583,7 +584,7 @@ def compute_global_regression(df_combined, embedding_cols, tsne_params, df_f1, r
583
  ("umap", umap.UMAP(
584
  n_components=N_COMPONENTS,
585
  random_state=42,
586
- n_neighbors=15,
587
  min_dist=0.1,
588
  metric='cosine'
589
  )),
@@ -1072,7 +1073,7 @@ def run_model(model_name):
1072
  ("umap", umap.UMAP(
1073
  n_components=N_COMPONENTS,
1074
  random_state=42,
1075
- n_neighbors=15,
1076
  min_dist=0.1,
1077
  metric='cosine'
1078
  )),
 
56
  N_COMPONENTS = 3
57
  TSNE_NEIGHBOURS = 15
58
  # WEIGHT_FACTOR = 0.05
59
+ N_NEIGHBOURS_UMAP = 50
60
 
61
 
62
  TOOLTIPS = """
 
584
  ("umap", umap.UMAP(
585
  n_components=N_COMPONENTS,
586
  random_state=42,
587
+ n_neighbors=N_NEIGHBOURS_UMAP,
588
  min_dist=0.1,
589
  metric='cosine'
590
  )),
 
1073
  ("umap", umap.UMAP(
1074
  n_components=N_COMPONENTS,
1075
  random_state=42,
1076
+ n_neighbors=N_NEIGHBOURS_UMAP,
1077
  min_dist=0.1,
1078
  metric='cosine'
1079
  )),