21 lines
453 B
YAML
21 lines
453 B
YAML
stages:
|
|
train:
|
|
deps:
|
|
- data/dataset/train.csv
|
|
- data/dataset/val.csv
|
|
- data/dataset/test.csv
|
|
outs:
|
|
- results/model.pt
|
|
params:
|
|
- train.data_len
|
|
- train.epochs
|
|
cmd: python src/main.py --train --data_len=${train.data_len} -e=${train.epochs}
|
|
test:
|
|
deps:
|
|
- data/dataset/test.csv
|
|
- data/dataset/test.csv
|
|
- data/dataset/test.csv
|
|
outs:
|
|
- results/results.csv
|
|
cmd: python src/main.py --test
|