From e587aa4b9599fb9c2890c03083b866b461b45a4a Mon Sep 17 00:00:00 2001
From: vijaysamula <samulavijay@gmail.com>
Date: Tue, 3 Jan 2023 11:54:12 +0100
Subject: [PATCH 1/2] Modify forest depth

---
 .github/workflows/cml.yaml | 19 +++++++++++++++++++
 train.py                   |  2 +-
 2 files changed, 20 insertions(+), 1 deletion(-)
 create mode 100644 .github/workflows/cml.yaml

diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
new file mode 100644
index 000000000..30e129243
--- /dev/null
+++ b/.github/workflows/cml.yaml
@@ -0,0 +1,19 @@
+name: CML
+on: [push]
+jobs:
+  train-and-report:
+    runs-on: ubuntu-latest
+    container: docker://ghcr.io/iterative/cml:0-dvc2-base1
+    steps:
+      - uses: actions/checkout@v3
+      - name: Train model
+        env:
+          REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          pip install -r requirements.txt
+          python train.py  # generate plot.png
+
+          # Create CML report
+          cat metrics.txt >> report.md
+          echo '![](./plot.png "Confusion Matrix")' >> report.md
+          cml comment create report.md
\ No newline at end of file
diff --git a/train.py b/train.py
index 415d9a9f9..f44634667 100644
--- a/train.py
+++ b/train.py
@@ -12,7 +12,7 @@
 y_test = np.genfromtxt("data/test_labels.csv")
 
 # Fit a model
-depth = 2
+depth = 5
 clf = RandomForestClassifier(max_depth=depth)
 clf.fit(X_train, y_train)
 

From 5e6431fbaa541d34fe6f6fc4caee3600f94a0598 Mon Sep 17 00:00:00 2001
From: vijaysamula <samulavijay@gmail.com>
Date: Tue, 3 Jan 2023 12:27:26 +0100
Subject: [PATCH 2/2] Modify forest depth 10

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index f44634667..883a4c456 100644
--- a/train.py
+++ b/train.py
@@ -12,7 +12,7 @@
 y_test = np.genfromtxt("data/test_labels.csv")
 
 # Fit a model
-depth = 5
+depth = 10
 clf = RandomForestClassifier(max_depth=depth)
 clf.fit(X_train, y_train)