schema: '2.0'
stages:
  split_data:
    cmd: python IUM_05-split.py
    deps:
    - path: data/beer_reviews.csv
      hash: md5
      md5: 50f6eec0d0fe78bc0f10e35edd271998
      size: 201644905
    outs:
    - path: beer_reviews_test.csv
      hash: md5
      md5: edbd0a7f05c59a0c0e936917f60e9b96
      size: 40632354
    - path: beer_reviews_train.csv
      hash: md5
      md5: 8c6877a26fef1542369bfae6b39d163c
      size: 162599343
  train_model:
    cmd: python IUM_05-model.py 10 32
    deps:
    - path: beer_reviews_train.csv
      hash: md5
      md5: 8c6877a26fef1542369bfae6b39d163c
      size: 162599343
    outs:
    - path: beer_review_sentiment_model.h5
      hash: md5
      md5: c126bd5d332a905262c66894585450e3
      size: 1950856
  predict:
    cmd: python IUM_05-predict.py
    deps:
    - path: beer_review_sentiment_model.h5
      hash: md5
      md5: c126bd5d332a905262c66894585450e3
      size: 1950856
    - path: beer_reviews_test.csv
      hash: md5
      md5: edbd0a7f05c59a0c0e936917f60e9b96
      size: 40632354
    outs:
    - path: beer_review_sentiment_predictions.csv
      hash: md5
      md5: 12a66fafb7f4d7d19eb0c4a90cc7d3ad
      size: 4814242