24 lines
490 B
YAML
24 lines
490 B
YAML
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
|