stages:
  prepare:
    cmd: download_dataset.sh
    
  train:
    cmd: python3 train_model.py
    deps:
      - data/Plant_1_Generation_data.csv.train
    outs:
      - model_out
    plots:
      - trend.png
    params:
      - batch_size
      - epochs