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/custom_data_loading.py - src/code/params.yml - src/code/training.py - src/data/processed/train outs: - src/models/ - logs/train_params.yml: cache: false metrics: - logs/train_metrics.csv: cache: false eval: cmd: python3 src/code/eval.py src/data/processed/test deps: - src/code/params.yml - src/code/custom_data_loading.py - src/code/eval_metric_calculation.py - src/code/eval.py - src/models/model.pth - src/data/processed/test metrics: - logs/test_metrics.csv: cache: false