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: aabaf1a407badf48c97b14a69b0072ea
|
|
size: 3420
|
|
params:
|
|
params.yaml:
|
|
train:
|
|
data_dir: data/processed
|
|
epochs: 5
|
|
batch_size: 256
|
|
learning_rate: 5e-05
|
|
outs:
|
|
- path: model.pt
|
|
hash: md5
|
|
md5: 8ead2a7cd52d70b359d3cdc3df5e43e3
|
|
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: 8a9a2e95a6b64e632a4f2feac62d294b
|
|
size: 1473
|
|
- path: model.pt
|
|
hash: md5
|
|
md5: 8ead2a7cd52d70b359d3cdc3df5e43e3
|
|
size: 102592994
|
|
params:
|
|
params.yaml:
|
|
evaluate:
|
|
data_dir: data/processed
|