iterative · sshtomar · Apr 8, 2023 · Apr 8, 2023 · Apr 8, 2023 · Apr 8, 2023
diff --git a/.github/workflows/cml.yaml 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.PERSONAL_ACCESS_TOKEN }}
+        run: |
+          pip install -r requirements.txt
+          python train.py  # generate plot.png
+
+          # Create CML report
+          cat metrics.txt >> report.md
+          echo '![](./plot confusion matrix.png "Confusion Matrix")' >> report.md
+          cml comment create report.md
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+# environment
+cml-env/
diff --git a/data/test_features.csv b/data/test_features.csv
diff --git a/data/train_features.csv b/data/train_features.csv
diff --git a/get_data.py b/get_data.py
@@ -16,4 +16,4 @@
 np.savetxt("data/train_features.csv", X_train)
 np.savetxt("data/test_features.csv", X_test)
 np.savetxt("data/train_labels.csv", y_train)
-np.savetxt("data/test_labels.csv", y_test)
+np.savetxt("data/test_labels.csv", y_test)
diff --git a/metrics.txt b/metrics.txt
@@ -0,0 +1 @@
+Accuracy: 0.864
diff --git a/plot.png b/plot.png
diff --git a/train.py b/train.py
@@ -12,7 +12,7 @@
 y_test = np.genfromtxt("data/test_labels.csv")
 
 # Fit a model
-depth = 2
+depth = 6
 clf = RandomForestClassifier(max_depth=depth)
 clf.fit(X_train, y_train)
 
@@ -25,4 +25,4 @@
 disp = ConfusionMatrixDisplay.from_estimator(
     clf, X_test, y_test, normalize="true", cmap=plt.cm.Blues
 )
-plt.savefig("plot.png")
+plt.savefig("plot confusion matrix.png")