schema: '2.0'
stages:
  test:
    cmd: python src/main.py --test
    deps:
    - path: data/dataset/test.csv
      md5: 9dd26ae5484ab058be21ae695098b9c6
      size: 10985891
    - path: data/dataset/test.csv
      md5: 9dd26ae5484ab058be21ae695098b9c6
      size: 10985891
    - path: data/dataset/test.csv
      md5: 9dd26ae5484ab058be21ae695098b9c6
      size: 10985891
    outs:
    - path: results/results.csv
      md5: 0a4ca4ee424a0668a2748987de74236a
      size: 815
  train:
    cmd: python src/main.py --train --data_len=200 -e=1
    deps:
    - path: data/dataset/test.csv
      md5: 9dd26ae5484ab058be21ae695098b9c6
      size: 10985891
    - path: data/dataset/train.csv
      md5: b4c5dfbc36c057a27137766197adbeca
      size: 89199212
    - path: data/dataset/val.csv
      md5: 64b7d344d7d4d7117b2cf9d7e31be771
      size: 11229062
    params:
      params.yaml:
        train.data_len: 200
        train.epochs: 1
    outs:
    - path: results/model.pt
      md5: c6a52e8320ccee83efb634f18c09a5c7
      size: 433319573