This project implements a sentiment analysis model using sentence embeddings for feature extraction. The model classifies tweets into three sentiment categories: positive, negative, and neutral. It employs a neural network built with PyTorch and uses the SentenceTransformers library to obtain sentence embeddings.
This repository provides a comprehensive pipeline for sentiment analysis:
- Data Preparation: Load and preprocess tweet data.
- Feature Extraction: Use SentenceTransformers to convert tweets into embeddings.
- Model Training: Train a neural network to classify sentiments.
- Evaluation: Assess model performance using accuracy metrics.
To run this code, you'll need to install the required packages. You can use pip to install the necessary libraries:
pip install numpy pandas torch sentence-transformers seaborn scikit-learn