diff --git a/.github/workflows/cml.yaml b/.github/workflows/cml.yaml new file mode 100644 index 000000000..f79f2aa1f --- /dev/null +++ b/.github/workflows/cml.yaml @@ -0,0 +1,29 @@ +name: your-workflow-name +on: [push] +jobs: + run: + runs-on: ubuntu-latest + # optionally use a convenient Ubuntu LTS + DVC + CML image + container: ghcr.io/iterative/cml:0-dvc2-base1 + steps: + - uses: actions/checkout@v3 + # may need to setup NodeJS & Python3 on e.g. self-hosted + # - uses: actions/setup-node@v3 + # with: + # node-version: '16' + # - uses: actions/setup-python@v4 + # with: + # python-version: '3.x' + - uses: iterative/setup-cml@v1 + - name: Train model + run: | + # Your ML workflow goes here + pip install -r requirements.txt + python train.py + - name: Write CML report + env: + REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }} + run: | + # Post reports as comments in GitHub PRs + cat results.txt >> 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