18 lines
337 B
YAML
18 lines
337 B
YAML
|
stages:
|
||
|
|
download_and_split:
|
||
|
|
cmd: python3 split_10.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
|
||
|
|
outs:
|
||
|
|
- Y_pred.txt
|