Movie Rating Predictor - The transformer

Many movie rating sites are in need of machine’s help these days. Due, to growing number of people reviewing movies, It’s hard for people to read each review and give it a rating. The aim here is to automate this process i.e machines should understand the language and automatically should output the rating based on reviews given by humans. Transformers are the state of the art models especially used in Natural Language Processing(NLP). So, I have trained an XLNet(with few cnns and dense layers on top of it) for predicting the rating.

Feel free to visit my github repo for deeper details.