stages:
  prepare:
    cmd: download_dataset.sh
    
  train:
    cmd: python3 train_model.py
    deps:
      - data/Plant_1_Generation_data.csv.train
    params:
      - batch_size
      - epochs
  
  eval:
    cmd: python3 eval_model.py