stages:
  split_data:
    cmd: python IUM_05-split.py
    deps:
      - data/beer_reviews.csv
    outs:
      - beer_reviews_train.csv
      - beer_reviews_test.csv

  train_model:
    cmd: python IUM_05-model.py 10 32
    deps:
      - beer_reviews_train.csv
    outs:
      - beer_review_sentiment_model.h5

  predict:
    cmd: python IUM_05-predict.py
    deps:
      - beer_review_sentiment_model.h5
      - beer_reviews_test.csv
    outs:
      - beer_review_sentiment_predictions.csv