14 lines
238 B
YAML
14 lines
238 B
YAML
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 |