This project implements a sentiment analysis model using PyTorch and the IMDB dataset from the Hugging Face datasets library. The model uses an LSTM architecture to classify movie reviews as either positive or negative.
- Loads the IMDB dataset from the Hugging Face datasets library
- Tokenizes the text data using the BERT tokenizer
- Trains a sentiment analysis model using PyTorch and LSTM
- Evaluates the model's performance on the test set
- Provides a Flask API for sentiment prediction
- Python 3.7 or higher
- PyTorch
- Hugging Face Transformers
- Hugging Face Datasets
- Flask
- Flask-CORS