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: 0a4ca4ee424a0668a2748987de74236a
|
||
|
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: c6a52e8320ccee83efb634f18c09a5c7
|
||
|
size: 433319573
|