17 lines
308 B
YAML
17 lines
308 B
YAML
|
stages:
|
||
|
|
prepare:
|
||
|
|
cmd: python create_data.py
|
||
|
|
deps:
|
||
|
|
- create_data.py
|
||
|
|
- KaggleV2-May-2016.csv
|
||
|
|
outs:
|
||
|
|
- train_dataset_dvc.csv
|
||
|
|
- test_dataset_dvc.csv
|
||
|
|
train:
|
||
|
|
cmd: python train_model.py
|
||
|
|
deps:
|
||
|
|
- train_dataset_dvc.csv
|
||
|
|
- test_dataset_dvc.csv
|
||
|
|
outs:
|
||
|
|
- logs_dvc.txt
|