stages:
  preprocess:
    cmd: python better_preprocess.py
    deps:
      - 25k_IMDb_movie_Dataset.csv
    outs:
      - data.csv

  train:
    cmd: python script5_3.py
    deps:
      - data.csv
    outs:
      - model.pth