Fix overall score incorrect

#7
by SmileXing - opened
Files changed (1) hide show
  1. app/backend/data_engine.py +1 -1
app/backend/data_engine.py CHANGED
@@ -129,7 +129,7 @@ class DataEngine:
129
  # set dataset default value to 0
130
  df = df.pivot(index=["model_name", "embd_dim", "embd_dtype", "group_name"], columns="dataset_name",
131
  values=["ndcg_at_10"]).fillna(0).stack(level=1).reset_index()
132
- # df = pd.merge(df, df_model, on=["model_name"], how="inner")
133
 
134
  # dataset_num_map = {}
135
  # grouped_dataset_count = df.groupby(["group_name"]).agg({
 
129
  # set dataset default value to 0
130
  df = df.pivot(index=["model_name", "embd_dim", "embd_dtype", "group_name"], columns="dataset_name",
131
  values=["ndcg_at_10"]).fillna(0).stack(level=1).reset_index()
132
+ df = pd.merge(df, df_dataset, on=["group_name","dataset_name"], how="inner")
133
 
134
  # dataset_num_map = {}
135
  # grouped_dataset_count = df.groupby(["group_name"]).agg({