Create cml.yaml

.github/workflows/cml.yaml

@@ -0,0 +1,31 @@
+name: model-training
+on: [push]
+jobs:
+  run:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-python@v4
+      - uses: iterative/setup-cml@v1
+      - name: Train model
+        env:
+          REPO_TOKEN: ${{ github.token }}
+        run: |
+          pip install -r requirements.txt
+          python train.py
+
+
+          echo "## MODEL METRICS" > report.md
+          cat metrics.txt >> report.md
+
+          echo "## Data Viz" >> report.md
+          echo 'edited' >> report.md
+          echo "![](./feature_importance.png)" >> report.md
+          echo "![](./residuals.png)" >> report.md
+
+          cml comment create report.md
+
+
+
+
+

requirements.txt

@@ -1,5 +1,5 @@
 pandas
-sklearn
+scikit-learn
 matplotlib
 seaborn
 numpy