stages:
  prepare_data:
    cmd: python ./create-dataset.py
    deps:
      - create-dataset.py
      - creditcard.csv
    outs:
      - data/creditcard.csv
      - data/X_train.csv
      - data/X_val.csv
      - data/X_test.csv
      - data/y_train.csv
      - data/y_val.csv
      - data/y_test.csv

  train_model:
    cmd: python ./train_model.py
    deps:
      - train_model.py
      - data/X_train.csv
      - data/X_val.csv
      - data/y_train.csv
      - data/y_val.csv
    outs:
      - model/model.keras

  evaluate_model:
    cmd: python ./predict.py
    deps:
      - predict.py
      - model/model.keras
      - data/X_test.csv
      - data/y_test.csv
    outs:
      - data/y_pred.csv