stages:
  split_data:
    cmd: python get_dataset.py
    deps:
      - Student_Performance.csv
    outs:
      - dataset.csv
      - df_train.csv
      - df_test.csv

  train_model:
    cmd: python train.py 30
    deps:
      - df_train.csv
      - df_test.csv
    outs:
      - model.keras