From 8c38bf4aa727adddbb953acb07bda34c6821f2a0 Mon Sep 17 00:00:00 2001
From: Paupiera <pau.piera.linde@gmail.com>
Date: Tue, 9 Jan 2024 20:19:50 +0100
Subject: [PATCH] add cml

---
 .github/cml.yaml | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 .github/cml.yaml

diff --git a/.github/cml.yaml b/.github/cml.yaml
new file mode 100644
index 0000000..cebd97d
--- /dev/null
+++ b/.github/cml.yaml
@@ -0,0 +1,21 @@
+name: train-my-model
+on: [push]
+jobs:
+  run:
+    runs-on: [ubuntu-latest]
+    steps:
+      - uses: actions/checkout@v2
+      - uses: iterative/setup-cml@v1
+      - name: Train model
+        run: |
+          pip install -r requirements.txt  # install dependencies
+          python train_cml.py  # run training
+      - name: Write report
+        env:
+          # this authenticates that the right permissions are in place
+          REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          # send all information to report.md that will be reported to us when the workflow finish
+          cat classification_report.txt >> report.md
+          cml-publish confusion_matrix.png --md >> report.md
+          cml-send-comment report.md