stages:
  train:
    deps:
    - data/dataset/train.csv
    - data/dataset/val.csv
    - data/dataset/test.csv
    outs:
    - results/model.pt
    params:
    - train.data_len
    - train.epochs
    cmd: python src/main.py --train --data_len=${train.data_len} -e=${train.epochs}
  test:
    deps:
    - data/dataset/test.csv
    - data/dataset/test.csv
    - data/dataset/test.csv
    outs:
    - results/results.csv
    cmd: python src/main.py --test