From bc46d90bc21512b9596a5dd44985a6cd3bd310ea Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Sat, 19 Aug 2023 17:16:23 +0000 Subject: [PATCH] Add markdown metric file for CML --- model_data/simulation_system/MLModel/Models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/model_data/simulation_system/MLModel/Models.py b/model_data/simulation_system/MLModel/Models.py index 7f21e548..137f2f20 100644 --- a/model_data/simulation_system/MLModel/Models.py +++ b/model_data/simulation_system/MLModel/Models.py @@ -116,6 +116,8 @@ class AutogluonModel: metrics_df = pd.DataFrame([performance]) metrics_df.to_csv(metrics_location / metric_filename) + markdown_filename = metric_filename.split(".")[0] + ".md" + metrics_df.to_markdown(metrics_location/ markdown_filename) return metrics_df