Spaces:
Running
Running
Set Dataset Default Value to 0
#4
by
SmileXing
- opened
app/backend/data_engine.py
CHANGED
@@ -120,13 +120,13 @@ class DataEngine:
|
|
120 |
}).reset_index()
|
121 |
|
122 |
pivot = grouped_model.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="group_name",
|
123 |
-
values=["ndcg_at_10"]).round(2)
|
124 |
|
125 |
# Rename columns
|
126 |
pivot.columns = list(
|
127 |
map(lambda x: f"{x[1].capitalize()} Average" if x[1] != 'text' else f"Average", pivot.columns))
|
128 |
|
129 |
-
pivot_dataset = df_result.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="dataset_name", values="ndcg_at_10")
|
130 |
|
131 |
df = pd.merge(df_model, pivot, on=["model_name", "embd_dim", "embd_dtype"])
|
132 |
df = pd.merge(df, pivot_dataset, on=["model_name", "embd_dim", "embd_dtype"])
|
|
|
120 |
}).reset_index()
|
121 |
|
122 |
pivot = grouped_model.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="group_name",
|
123 |
+
values=["ndcg_at_10"]).round(2).fillna(0)
|
124 |
|
125 |
# Rename columns
|
126 |
pivot.columns = list(
|
127 |
map(lambda x: f"{x[1].capitalize()} Average" if x[1] != 'text' else f"Average", pivot.columns))
|
128 |
|
129 |
+
pivot_dataset = df_result.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="dataset_name", values="ndcg_at_10").fillna(0)
|
130 |
|
131 |
df = pd.merge(df_model, pivot, on=["model_name", "embd_dim", "embd_dtype"])
|
132 |
df = pd.merge(df, pivot_dataset, on=["model_name", "embd_dim", "embd_dtype"])
|