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