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