Update cml.yaml

elleobrien · Dec 29, 2023 · ae733be · ae733be
1 parent 4798e9e
commit ae733be
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
@@ -21,5 +21,13 @@ jobs:
           pip install -r requirements.txt
           python train.py
 
-      echo "MODEL METRICS"
-      cat metrics.txt
+      echo "## MODEL METRICS" > report.md
+      cat metrics.txt >> report.md
+
+      echo "## Data Viz" >> report.md
+      cml-publish feature_importance.png --md >> report.md
+      cml-publish residuals.png --md >> report.md
+
+      cml-send-comment report.md
+
+