Fill Nan Before Avg (#5)

- feat - fill na before avg (115fd813844199daf7ec4481da84211f3313b23a)


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

app/backend/data_engine.py

@@ -105,6 +105,10 @@ class DataEngine:
         df_model = pd.DataFrame(models_list)
 
         df = pd.merge(df_result, df_dataset, on=["dataset_name"], how="inner")
+
+        # set dataset default value to 0
+        df = df.pivot(index=["model_name", "embd_dim", "embd_dtype", "group_name"], columns="dataset_name",
+                 values=["ndcg_at_10"]).fillna(0).stack(level=1).reset_index()
         # df = pd.merge(df, df_model, on=["model_name"], how="inner")
 
         # dataset_num_map = {}