Amazone Review Rating Prediction with BERT4REC(Transformer4REC)

BERT4REC is specialized at predicting the next item that a user most likely to buy. Objective of BERT4REC is MLM(Masked Language Modeling) meaning that it can't predict numeric value.

In this project, I modify BERT4REC to make it working as a regression model sothat the model can predict the rating of a product.

Quick Start

python main.py --data_dir "../data" --batch_size 256 --epoch_size 7

References

• https://arxiv.org/abs/1904.06690
• https://keras.io/examples/structured_data/movielens_recommendations_transformers/