IMDb Sentiment Neural Network
Neural network sentiment classifier trained on IMDb reviews.
Features
- TF-IDF Vectorization
- Feedforward Neural Network
- PyTorch
- GitHub Actions CI/CD
- Automatic Hugging Face Deployment
Dataset
IMDb movie review dataset.
Install
pip install -r requirements.txt
Train Model
Predict
CI/CD Pipeline
On every push to main:
- Install dependencies
- Train model
- Save artifacts
- Upload to Hugging Face Hub automatically
Hugging Face
Artifacts deployed automatically using GitHub Actions.
