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
|
|
|
|
|
outs:
|
|
|
|
|
- model_out
|
|
|
|
|
plots:
|
|
|
|
|
- trend.png
|
|
|
|
|
params:
|
2022-06-05 22:19:45 +02:00
|
|
|
- batch_size
|
|
|
|
|
- epochs
|
