IMDB Movie Review Sentiment Analysis
Overview
This project performs binary sentiment classification on IMDB movie reviews (positive/negative).
Model & Method
- Feature: TF-IDF
- Model: 2-layer fully connected network
- Framework: PyTorch
Performance
- Accuracy: 88%
Files
- train.py: training script
- imdb_balanced_10k.csv: dataset
- requirements.txt: dependencies
Run
pip install -r requirements.txt
python train.py