18 lines
322 B
YAML
18 lines
322 B
YAML
|
stages:
|
||
|
download_and_split:
|
||
|
cmd: python3 split_10.py
|
||
|
deps:
|
||
|
- heart_failure_clinical_records_dataset.csv
|
||
|
- split_10.py
|
||
|
outs:
|
||
|
- train.csv
|
||
|
- valid.csv
|
||
|
- test.csv
|
||
|
train_model:
|
||
|
cmd: python3 train_10.py
|
||
|
deps:
|
||
|
- train.csv
|
||
|
- valid.csv
|
||
|
- test.csv
|
||
|
outs:
|
||
|
- accuracy.txt
|