de-Rodrigo commited on
Commit
576d618
·
1 Parent(s): f5f7066

Fix Reduction Pipeline

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -859,7 +859,8 @@ def run_model(model_name):
859
  df_all = {}
860
  # Real
861
  df_real_proj = embeddings["real"].copy()
862
- proj_real = reducer_real.named_steps["pca"].transform(df_real_proj[embedding_cols].values)
 
863
  for i in range(proj_real.shape[1]):
864
  df_real_proj[f'PC{i+1}'] = proj_real[:, i]
865
  df_all["real"] = df_real_proj
@@ -867,7 +868,7 @@ def run_model(model_name):
867
  # Synthetic
868
  if "synthetic" in embeddings:
869
  df_synth_proj = embeddings["synthetic"].copy()
870
- proj_synth = reducer_real.named_steps["pca"].transform(df_synth_proj[embedding_cols].values)
871
  for i in range(proj_synth.shape[1]):
872
  df_synth_proj[f'PC{i+1}'] = proj_synth[:, i]
873
  df_all["synthetic"] = df_synth_proj
@@ -875,7 +876,7 @@ def run_model(model_name):
875
  # Pretrained
876
  if "pretrained" in embeddings:
877
  df_pretr_proj = embeddings["pretrained"].copy()
878
- proj_pretr = reducer_real.named_steps["pca"].transform(df_pretr_proj[embedding_cols].values)
879
  for i in range(proj_pretr.shape[1]):
880
  df_pretr_proj[f'PC{i+1}'] = proj_pretr[:, i]
881
  df_all["pretrained"] = df_pretr_proj
 
859
  df_all = {}
860
  # Real
861
  df_real_proj = embeddings["real"].copy()
862
+ # proj_real = reducer_real.named_steps["pca"].transform(df_real_proj[embedding_cols].values)
863
+ proj_real = reducer_real.transform(df_real_proj[embedding_cols].values)
864
  for i in range(proj_real.shape[1]):
865
  df_real_proj[f'PC{i+1}'] = proj_real[:, i]
866
  df_all["real"] = df_real_proj
 
868
  # Synthetic
869
  if "synthetic" in embeddings:
870
  df_synth_proj = embeddings["synthetic"].copy()
871
+ proj_synth = reducer_real.transform(df_synth_proj[embedding_cols].values)
872
  for i in range(proj_synth.shape[1]):
873
  df_synth_proj[f'PC{i+1}'] = proj_synth[:, i]
874
  df_all["synthetic"] = df_synth_proj
 
876
  # Pretrained
877
  if "pretrained" in embeddings:
878
  df_pretr_proj = embeddings["pretrained"].copy()
879
+ proj_pretr = reducer_real.transform(df_pretr_proj[embedding_cols].values)
880
  for i in range(proj_pretr.shape[1]):
881
  df_pretr_proj[f'PC{i+1}'] = proj_pretr[:, i]
882
  df_all["pretrained"] = df_pretr_proj