stages:
prepare_data:
cmd: ./download.sh
deps:
- download.sh
outs:
- data/raw
train:
cmd: python train.py
- train.py
- model.pth
test:
cmd: python test.py
- test.py
- predictions.csv