q275343119 commited on
Commit
115fd81
·
1 Parent(s): 7e203f8

feat - fill na before avg

Browse files
Files changed (1) hide show
  1. app/backend/data_engine.py +4 -0
app/backend/data_engine.py CHANGED
@@ -105,6 +105,10 @@ class DataEngine:
105
  df_model = pd.DataFrame(models_list)
106
 
107
  df = pd.merge(df_result, df_dataset, on=["dataset_name"], how="inner")
 
 
 
 
108
  # df = pd.merge(df, df_model, on=["model_name"], how="inner")
109
 
110
  # dataset_num_map = {}
 
105
  df_model = pd.DataFrame(models_list)
106
 
107
  df = pd.merge(df_result, df_dataset, on=["dataset_name"], how="inner")
108
+
109
+ # set dataset default value to 0
110
+ df = df.pivot(index=["model_name", "embd_dim", "embd_dtype", "group_name"], columns="dataset_name",
111
+ values=["ndcg_at_10"]).fillna(0).stack(level=1).reset_index()
112
  # df = pd.merge(df, df_model, on=["model_name"], how="inner")
113
 
114
  # dataset_num_map = {}