de-Rodrigo commited on
Commit
7ff9943
·
1 Parent(s): 0f93112

Provide Features

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -602,7 +602,8 @@ def compute_global_regression(df_combined, embedding_cols, tsne_params, df_f1, r
602
  # silhouette = -1
603
 
604
  df_heat = pd.read_csv(f"data/heatmaps_idefics2-patient.csv")
605
- feature_options = [col for col in df_heat.columns if col != "name"]
 
606
 
607
  silhouette_vals = []
608
  for feature in feature_options:
 
602
  # silhouette = -1
603
 
604
  df_heat = pd.read_csv(f"data/heatmaps_idefics2-patient.csv")
605
+ # feature_options = [col for col in df_heat.columns if col != "name"]
606
+ feature_options = ["grades", "table_pos", "v_density", "v_info_blocks", "layout", "columns", "table_grid", "header_badge"]
607
 
608
  silhouette_vals = []
609
  for feature in feature_options: