stages:
  process_data:
    cmd: python3 src/code/make_dataset.py src/data/raw/nyu_depth_v2_labeled.mat src/data/raw/splits.mat
      src/data/processed
    deps:
    - src/code/make_dataset.py
    - src/data/raw/nyu_depth_v2_labeled.mat
    - src/data/raw/splits.mat
    outs:
    - src/data/processed/
  train:
    cmd: python3 src/code/training.py src/data/processed/train
    deps:
    - src/code/training.py
    - src/data/processed/train
    outs:
    - src/models/
  eval:
    cmd: python3 src/code/eval.py src/data/processed/test
    deps:
    - src/code/eval.py
    - src/models/model.pth
    - src/data/processed/test
    outs:
    - src/eval/