24 lines
322 B
YAML
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
|