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