Set Dataset Default Value to 0

#4
Files changed (1) hide show
  1. app/backend/data_engine.py +2 -2
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"])