stages:
  train:
    cmd: python train.py --train=true --save_model=true
  test:
    cmd: python train_model.py