stages:
  prepare_datasets:
    cmd: python prepare_datasets.py X_train.csv X_test.csv y_train.csv y_test.csv
    deps:
    - data/bodyPerformance.csv
    - prepare_datasets.py
    outs:
    - X_test.csv
    - X_train.csv
    - y_test.csv
    - y_train.csv
  train:
    cmd: python train.py model.pth
    deps:
    - X_train.csv
    - train.py
    - y_train.csv
    outs:
    - model.pth