ium_464915/dvc.yaml
2024-05-29 21:04:36 +02:00

24 lines
322 B
YAML

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