39 lines
1.0 KiB
Plaintext
39 lines
1.0 KiB
Plaintext
schema: '2.0'
|
|
stages:
|
|
test:
|
|
cmd: python src/main.py --test
|
|
deps:
|
|
- path: data/dataset/test.csv
|
|
md5: 9dd26ae5484ab058be21ae695098b9c6
|
|
size: 10985891
|
|
- path: data/dataset/test.csv
|
|
md5: 9dd26ae5484ab058be21ae695098b9c6
|
|
size: 10985891
|
|
- path: data/dataset/test.csv
|
|
md5: 9dd26ae5484ab058be21ae695098b9c6
|
|
size: 10985891
|
|
outs:
|
|
- path: results/results.csv
|
|
md5: b8ca3bc842a11fbb9ba0538a78acccec
|
|
size: 815
|
|
train:
|
|
cmd: python src/main.py --train --data_len=200 -e=1
|
|
deps:
|
|
- path: data/dataset/test.csv
|
|
md5: 9dd26ae5484ab058be21ae695098b9c6
|
|
size: 10985891
|
|
- path: data/dataset/train.csv
|
|
md5: b4c5dfbc36c057a27137766197adbeca
|
|
size: 89199212
|
|
- path: data/dataset/val.csv
|
|
md5: 64b7d344d7d4d7117b2cf9d7e31be771
|
|
size: 11229062
|
|
params:
|
|
params.yaml:
|
|
train.data_len: 200
|
|
train.epochs: 1
|
|
outs:
|
|
- path: results/model.pt
|
|
md5: 5329e68ba53105d7060a5bae29bf4405
|
|
size: 433319573
|