Set model name in to_dataframe_all

app.py

@@ -52,11 +52,10 @@ def load_data(result_path) -> pd.DataFrame:
 def load_result(model_id):
     result_path = get_result_path_from_model(model_id, latest_result_path_per_model)
     data = load_data(result_path)
-    model_name = data.get("model_name", "Model")
     df = to_dataframe_all(data)
     result = [
-        to_vertical(df, model_name),
-        to_vertical(to_dataframe_results(df), model_name)
+        to_vertical(df),
+        to_vertical(to_dataframe_results(df))
     ]
     return result
 
@@ -65,8 +64,9 @@ def to_dataframe(data):
     return pd.DataFrame.from_records([data])
 
 
-def to_vertical(df, model_name):
-    df = df.iloc[0].rename_axis("Parameters").rename(model_name).to_frame()  # .reset_index()
+def to_vertical(df):
+    # df = df.iloc[0].rename_axis("Parameters").rename(model_name).to_frame()  # .reset_index()
+    df = df.iloc[0].rename_axis("Parameters").to_frame()  # .reset_index()
     df.index = df.index.str.join(".")
     return df
 
@@ -75,6 +75,7 @@ def to_dataframe_all(data):
     df = pd.json_normalize([{key: value for key, value in data.items() if key not in EXCLUDED_KEYS}])
     # df.columns = df.columns.str.split(".")  # .split return a list instead of a tuple
     df.columns = list(map(lambda x: tuple(x.split(".")), df.columns))
+    df.index = [data.get("model_name", "Model")]
     return df
 
 #