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