de-Rodrigo commited on
Commit
b52a288
·
1 Parent(s): 39b0cf3

Adjust UMAP Hyper Parameters

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -57,6 +57,7 @@ N_COMPONENTS = 3
57
  TSNE_NEIGHBOURS = 15
58
  # WEIGHT_FACTOR = 0.05
59
  N_NEIGHBOURS_UMAP = 50
 
60
 
61
 
62
  TOOLTIPS = """
@@ -585,7 +586,7 @@ def compute_global_regression(df_combined, embedding_cols, tsne_params, df_f1, r
585
  n_components=N_COMPONENTS,
586
  random_state=42,
587
  n_neighbors=N_NEIGHBOURS_UMAP,
588
- min_dist=0.1,
589
  metric='cosine'
590
  )),
591
  ("rel_scaler", RelativeScaler())
@@ -1074,7 +1075,7 @@ def run_model(model_name):
1074
  n_components=N_COMPONENTS,
1075
  random_state=42,
1076
  n_neighbors=N_NEIGHBOURS_UMAP,
1077
- min_dist=0.1,
1078
  metric='cosine'
1079
  )),
1080
  ("rel_scaler", RelativeScaler())
 
57
  TSNE_NEIGHBOURS = 15
58
  # WEIGHT_FACTOR = 0.05
59
  N_NEIGHBOURS_UMAP = 50
60
+ MIN_DIST_UMAP = 0.5
61
 
62
 
63
  TOOLTIPS = """
 
586
  n_components=N_COMPONENTS,
587
  random_state=42,
588
  n_neighbors=N_NEIGHBOURS_UMAP,
589
+ min_dist=MIN_DIST_UMAP,
590
  metric='cosine'
591
  )),
592
  ("rel_scaler", RelativeScaler())
 
1075
  n_components=N_COMPONENTS,
1076
  random_state=42,
1077
  n_neighbors=N_NEIGHBOURS_UMAP,
1078
+ min_dist=MIN_DIST_UMAP,
1079
  metric='cosine'
1080
  )),
1081
  ("rel_scaler", RelativeScaler())