19 lines
354 B
YAML
19 lines
354 B
YAML
stages:
|
|
download_and_split:
|
|
cmd: python3 create.py
|
|
deps:
|
|
- healthcare-dataset-stroke-data.csv
|
|
- create.py
|
|
outs:
|
|
- data_train.csv
|
|
- data_test.csv
|
|
- data_val.csv
|
|
train_model:
|
|
cmd: python3 train-dvc.py
|
|
deps:
|
|
- data_train.csv
|
|
- data_test.csv
|
|
- data_val.csv
|
|
- train-dvc.py
|
|
outs:
|
|
- Y_pred.txt |