2021-06-12 11:39:51 +02:00
|
|
|
stages:
|
|
|
|
|
download_and_split:
|
2021-06-12 11:49:08 +02:00
|
|
|
cmd: python3 create.py
|
2021-06-12 11:39:51 +02:00
|
|
|
deps:
|
|
|
|
|
- healthcare-dataset-stroke-data.csv
|
2021-06-12 12:00:15 +02:00
|
|
|
- create.py
|
2021-06-12 11:39:51 +02:00
|
|
|
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
|
2021-06-12 12:03:04 +02:00
|
|
|
- train-dvc.py
|
2021-06-12 11:39:51 +02:00
|
|
|
outs:
|
|
|
|
|
- Y_pred.txt
|
