stages:
  prepare:
    cmd: data.py
    deps:
    - data.py
    - BPD_Part_1_Victim_Based_Crime_Data.csv
    outs:
    - baltimore_test.csv
    - baltimore_train.csv
    - baltimore_dev.csv
  train:
    cmd: ium_train.py
    deps:
    - ium_train.py
    - baltimore_train.csv
    outs:
    - baltimore.zip