dvc.lock

schema: '2.0'
stages:
  get-data:
    cmd: python src/get_data.py --output_folder outs/data
    deps:
    - path: src/get_data.py
      md5: c7452de42878b5e9e672d786d48e7c53
      size: 998
    params:
      params.yaml:
        data:
          data_date: 2022/08/08
          output_folder: outs/data
          metrics_file: outs/data_metrics.json
    outs:
    - path: outs/data
      md5: 0223fa49fdf31fd4bb2f99cf8322f401.dir
      size: 136
      nfiles: 1
    - path: outs/data_metrics.json
      md5: 514e1d4b4da8fdb6174caca07f91d78c
      size: 43
  split-data:
    cmd: python src/split_data.py outs/data outs/split 0.1 outs/split_metrics.json
    deps:
    - path: outs/data
      md5: 7e11ff6b279d54d6aa1823277f4caf5c.dir
      size: 8087
      nfiles: 153
    - path: src/split_data.py
      md5: 4257c1dd971b77407f5e86af0e0edd4d
      size: 2177
    outs:
    - path: outs/split
      md5: 12e0041446108ced396e10623e115999.dir
      size: 12089
      nfiles: 2
    - path: outs/split_metrics.json
      md5: ca4b6423c8624682cc4858877707bb77
      size: 166
  compute-data-metrics:
    cmd: python src/compute_data_metrics.py --input_folder outs/data --output_metrics_file
      outs/data_metrics.json
    deps:
    - path: outs/data
      md5: ee718b82fa4b349030b0d118f6ad4eb0.dir
      size: 7726
      nfiles: 149
    - path: src/compute_data_metrics.py
      md5: 2667ba0ed4ea950a2c77df60a2921429
      size: 566
    outs:
    - path: outs/data_metrics.json
      md5: b47806515a30f27fc1bb228bc1389a1f
      size: 63
  train:
    cmd: python src/train.py outs/split outs/train
    deps:
    - path: outs/split
      md5: 12e0041446108ced396e10623e115999.dir
      size: 12089
      nfiles: 2
    - path: src/train.py
      md5: 88cd88d5ff2ff67ee5764e24998f7d6f
      size: 2522
    params:
      params.yaml:
        data.labels:
        - data-sync
        - experiments
        - plots
        train:
          output_folder: outs/train
          metrics_folder: outs/train_metrics
          pretrained_model: MoritzLaurer/xtremedistil-l6-h256-mnli-fever-anli-ling-binary
          epochs: 8
    outs:
    - path: outs/train
      md5: 2d6f31d634d42ab9996d9cac216aff06.dir
      size: 51992747
      nfiles: 7
    - path: outs/train_metrics
      md5: 82d8979c72e4a138266128b85fdea5de.dir
      size: 1652
      nfiles: 10
    - path: outs/train_metrics.json
      md5: 29e927a8537684c88395bd3a6fee69fe
      size: 347
  eval:
    cmd: python src/eval.py outs/split/val.json outs/train outs/eval
    deps:
    - path: outs/split/val.json
      md5: 9f3681b095600e87d0bc922ecc902cc3
      size: 1216
    - path: outs/train
      md5: 2d6f31d634d42ab9996d9cac216aff06.dir
      size: 51992747
      nfiles: 7
    - path: src/eval.py
      md5: f733df5cfaaef682afcc65babff16c06
      size: 978
    - path: src/inference.py
      md5: 0c3515e93bb2377cc71dd6086c9d1010
      size: 605
    params:
      params.yaml:
        data.labels:
        - data-sync
        - experiments
        - plots
    outs:
    - path: outs/eval/plots/confusion_matrix.json
      md5: 7ad7650d9b00d4ae671f04ba20889b79
      size: 1274
  fit:
    cmd: python src/fit.py outs/data outs/fit outs/fit_metrics
    deps:
    - path: outs/data
      md5: 0223fa49fdf31fd4bb2f99cf8322f401.dir
      size: 136
      nfiles: 1
    - path: src/fit.py
      md5: ef5f1cb19e0a3d286c43877e6b459620
      size: 947
    params:
      params.yaml:
        fit:
          output_folder: outs/fit
          metrics_folder: outs/fit_metrics
          metrics_file: outs/fit_metrics.json
          epochs: 8
    outs:
    - path: outs/fit
      md5: d751713988987e9331980363e24189ce.dir
      size: 0
      nfiles: 0
    - path: outs/fit_metrics
      md5: d751713988987e9331980363e24189ce.dir
      size: 0
      nfiles: 0
    - path: outs/fit_metrics.json
      md5: eba383f3de32d0a53c5962f4119e6dcf
      size: 23