tensorflow==2.8
sklearn
pandas
matplotlib
tensorflow_hub
The data set is from kaggle. The original authors of this dataset are Maas, Andrew L. and Daly, Raymond E. and Pham, Peter T. and Huang, Dan and Ng, Andrew Y. and Potts, Christopher from Learning Word Vectors for Sentiment Analysis, Proceedings of the 49th Annual Meeting of the Association for Computational Linguistics: Human Language Technologies, Association for Computational Linguistics, 142-150, 2011.
You can also use tfds.text.IMDBReviews.
We build a model to classify the reiviews of a movie. My goal is to use tfrecords for faster training.