dvc.lock

schema: '2.0'
stages:
  featurize:
    cmd: python3 src/featurize.py data/raw data/featurize
    deps:
    - path: data/raw
      md5: 9d3d74e16f896a88df4f97f9755bb4b0.dir
      size: 305179520
      nfiles: 5
    - path: src/featurize.py
      md5: c94a3fff61a5c2f572b027ce9c6d07c0
      size: 2022
    params:
      params.yaml:
        featurize.max_row_limit: 1000000
        featurize.window_size: 8
    outs:
    - path: data/featurize
      md5: 96dc364478248eea5f36cdb343cb0863.dir
      size: 83232709
      nfiles: 1
  train:
    cmd: python3 src/train.py data/featurize data/predictions
    deps:
    - path: data/featurize
      md5: 96dc364478248eea5f36cdb343cb0863.dir
      size: 83232709
      nfiles: 1
    - path: src/model.py
      md5: 39e8ead6259cd85a65a3e9e73ce9b70f
      size: 3742
    - path: src/train.py
      md5: 974d89f91c2f614d382e671e72ab5292
      size: 2185
    params:
      params.yaml:
        featurize.window_size: 8
        train.batch_size: 10000
        train.epochs: 100
        train.hidden_size: 20
    outs:
    - path: data/predictions
      md5: 7bde82a8b14a9e6d4a39021c3437d72b.dir
      size: 1374793
      nfiles: 1
  validate:
    cmd: python3 src/validate.py data/featurize data/predictions metrics.json
    deps:
    - path: data/featurize
      md5: 96dc364478248eea5f36cdb343cb0863.dir
      size: 83232709
      nfiles: 1
    - path: data/predictions
      md5: 7bde82a8b14a9e6d4a39021c3437d72b.dir
      size: 1374793
      nfiles: 1
    - path: src/validate.py
      md5: 313ee87dc6cd76edf73d77ea95526f44
      size: 2147
    params:
      params.yaml:
        featurize.window_size: 8
    outs:
    - path: metrics.json
      md5: f68b5abe81d2548d6d369a202a4a6815
      size: 92