polishin summary report

Author: Rabah Abdul Khalek

giskard_vision/landmark_detection/tests/report.py

@@ -114,8 +114,14 @@ def to_dataframe(self, summary: Optional[bool] = False):
                     "Best(prediction_fail_rate)",
                 ]
             ]
+            df.sort_values(["criteria", "model"], ignore_index=True)
+            return df.pivot(
+                columns="model",
+                values=["Best(prediction_time)", "Best(prediction_fail_rate)", "Best(metric_value)"],
+                index="criteria",
+            )
 
-        return df.sort_values(["criteria", "model"], ignore_index=True)
+        return df.sort_values(["criteria", "model"], ignore_index=True).drop(["group"], axis=1, errors="ignore")
 
     def to_markdown(self, summary: Optional[bool] = False, filename: Optional[str] = None):
         """
@@ -136,6 +142,9 @@ def to_markdown(self, summary: Optional[bool] = False, filename: Optional[str] =
 
         df = self.to_dataframe(summary=summary)
 
+        if summary:
+            df.columns = [" - ".join(col).strip() for col in df.columns.values]
+
         df.to_markdown(filename)
 
     def to_json(self, filename: Optional[str] = None):