15 lines
253 B
YAML
15 lines
253 B
YAML
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 |