2022-06-05 21:58:57 +02:00
|
|
|
stages:
|
|
|
|
|
prepare:
|
2022-06-05 22:19:45 +02:00
|
|
|
cmd: download_dataset.sh
|
2022-06-05 21:58:57 +02:00
|
|
|
|
|
|
|
|
train:
|
|
|
|
|
cmd: python3 train_model.py
|
|
|
|
|
deps:
|
|
|
|
|
- data/Plant_1_Generation_data.csv.train
|
|
|
|
|
params:
|
2022-06-05 22:19:45 +02:00
|
|
|
- batch_size
|
2022-06-05 22:25:53 +02:00
|
|
|
- epochs
|
|
|
|
|
|
|
|
|
|
eval:
|
2022-06-05 22:28:52 +02:00
|
|
|
cmd: python3 eval_model.py
|
