stages:
  prepare:
    cmd: python3 ml_prepare.py
    deps:
      - training_data.csv
      - test_data.csv
      - ml_prepare.py
  training:
    cmd: python3 ml_training.py 15
    deps:
      - ml_training.py