Skip to content

Update train.py

Update train.py #40

Workflow file for this run

name: model-training
on: [push]
jobs:
run:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/setup-python@v4
- uses: actions/setup-node@v1
with:
node-version: '16'
- uses: iterative/setup-cml@v1
- name: Train model
env:
DVC_STUDIO_TOKEN: ${{ secrets.DVC_STUDIO_TOKEN }}
REPO_TOKEN: ${{ secrets.GITHUB_TOKEN }}
run: |
pip install -r requirements.txt
cml ci --user-email=cml-bot@plexico.space --user-name=cml[bot]
python get_data.py
python train.py
cat metrics.txt >> report.md
echo "![](./plot.png)" >> report.md
cml comment create report.md