Set Dataset Default Value to 0 (#4)

- feat - set dataset default value to 0 (ce857b60d769f07202f915dde4d122075336bd10)


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

app/backend/data_engine.py

@@ -120,13 +120,13 @@ class DataEngine:
         }).reset_index()
 
         pivot = grouped_model.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="group_name",
-                                    values=["ndcg_at_10"]).round(2)
+                                    values=["ndcg_at_10"]).round(2).fillna(0)
 
         # Rename columns
         pivot.columns = list(
             map(lambda x: f"{x[1].capitalize()} Average" if x[1] != 'text' else f"Average", pivot.columns))
 
-        pivot_dataset = df_result.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="dataset_name", values="ndcg_at_10")
+        pivot_dataset = df_result.pivot(index=["model_name", "embd_dim", "embd_dtype"], columns="dataset_name", values="ndcg_at_10").fillna(0)
 
         df = pd.merge(df_model, pivot, on=["model_name", "embd_dim", "embd_dtype"])
         df = pd.merge(df, pivot_dataset, on=["model_name", "embd_dim", "embd_dtype"])