stages:
  prepare_data:
    cmd: ./download.sh
    deps:
      - download.sh
    outs:
      - data/raw

  train:
    cmd: python train.py
    deps:
      - train.py
      - data/raw
    outs:
      - model.pth

  test:
    cmd: python test.py
    deps:
      - test.py
      - model.pth
    outs:
      - predictions.csv