69 lines
1.5 KiB
Plaintext
69 lines
1.5 KiB
Plaintext
schema: '2.0'
|
|
stages:
|
|
prepare:
|
|
cmd: python prepare.py
|
|
deps:
|
|
- path: prepare.py
|
|
hash: md5
|
|
md5: f0fd9944ebe22e1af020f031847991ed
|
|
size: 2992
|
|
params:
|
|
params.yaml:
|
|
prepare:
|
|
data_dir: data/raw
|
|
save_dir: data/processed
|
|
train_valid_split:
|
|
- 0.7
|
|
- 0.3
|
|
random_seed: 0
|
|
outs:
|
|
- path: data/processed
|
|
hash: md5
|
|
md5: a47a61b2a1709f487fd286e9c54b89fc.dir
|
|
size: 295124946
|
|
nfiles: 60001
|
|
train:
|
|
cmd: python train.py
|
|
deps:
|
|
- path: data/processed
|
|
hash: md5
|
|
md5: a47a61b2a1709f487fd286e9c54b89fc.dir
|
|
size: 295124946
|
|
nfiles: 60001
|
|
- path: train.py
|
|
hash: md5
|
|
md5: d32feb4bad10a201fdde2b6424238d16
|
|
size: 3401
|
|
params:
|
|
params.yaml:
|
|
train:
|
|
data_dir: data/processed
|
|
epochs: 5
|
|
batch_size: 256
|
|
learning_rate: 5e-05
|
|
outs:
|
|
- path: model.pt
|
|
hash: md5
|
|
md5: a0823af18fbf58ff562d804a02dca7d2
|
|
size: 102592994
|
|
evaluate:
|
|
cmd: python evaluate.py
|
|
deps:
|
|
- path: data/processed
|
|
hash: md5
|
|
md5: a47a61b2a1709f487fd286e9c54b89fc.dir
|
|
size: 295124946
|
|
nfiles: 60001
|
|
- path: evaluate.py
|
|
hash: md5
|
|
md5: 8eb9acfdc80a5ca6b7bd782643a0997b
|
|
size: 1483
|
|
- path: model.pt
|
|
hash: md5
|
|
md5: a0823af18fbf58ff562d804a02dca7d2
|
|
size: 102592994
|
|
params:
|
|
params.yaml:
|
|
evaluate:
|
|
data_dir: data/processed
|