Create cml.yaml

Abhishek-N-D · Dec 26, 2023 · ebe0835 · ebe0835
1 parent 6e0ee0a
commit ebe0835
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml
@@ -0,0 +1,17 @@
+name: model-wine-quality
+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
+
+          echo "MODEL METRICS"
+          cat metrics.txt