stages:
  prepare_data:
    cmd: python3 clean_and_split_data.py
    deps:
    - crime_conv.csv
    outs:
    - crime_dev.csv
    - crime_test.csv
    - crime_train.csv
  train:
    cmd: python3 train_model.py
    deps:
    - crime_dev.csv
    - crime_test.csv
    - crime_train.csv
    outs:
    - model/saved_model.pb