Spaces:
Running
Running
Fix overall score incorrect (#7)
Browse files- fix - overall score incorrect (4f69af1f59f229f066ba6566b5ea6c872f2f6ba5)
Co-authored-by: Y <[email protected]>
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 |
-
|
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({
|