fzliu q275343119 commited on
Commit
3b4b828
·
verified ·
1 Parent(s): 5f8ba39

Fix overall score incorrect (#7)

Browse files

- fix - overall score incorrect (4f69af1f59f229f066ba6566b5ea6c872f2f6ba5)


Co-authored-by: Y <[email protected]>

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({