diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml new file mode 100644 index 000000000..4ff2de790 --- /dev/null +++ b/.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 + + + + + diff --git a/requirements.txt b/requirements.txt index b4a3372bd..3115634f5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ pandas -sklearn +scikit-learn matplotlib seaborn numpy