This project implements sentiment analysis for customer reviews using a fine-tuned DistilBERT model.
- Fine-tuning of pre-trained transformer model on IMDB dataset
- Sentiment classification (positive/negative)
- Visualization of sentiment distribution
- Analysis of customer reviews
- Install dependencies:
pip install -r requirements.txt
- Run the Jupyter notebook:
jupyter notebook sentiment_analysis_py.ipynb
The model is not included in this repository due to size constraints. To use this project:
- Run the notebook to train the model.
- Save the trained model.
- Replace the model path in the notebook.