iumKC/dvc.yaml

5 lines
111 B
YAML
Raw Normal View History

2024-05-26 14:02:37 +02:00
stages:
train:
cmd: python train.py --train=true --save_model=true
test:
cmd: python train_model.py