Fine-tuned all-MiniLM-L6-v2 for IMDB Sentiment Classification
This is a sentence-transformer model (sentence-transformers/all-MiniLM-L6-v2) fine-tuned for sentiment classification on the IMDB movie reviews dataset. It classifies text as either positive (LABEL_1) or negative (LABEL_0).
Model Description
This model was fine-tuned as part of a university assignment to demonstrate the effectiveness of transfer learning. By adapting a pre-trained sentence encoder to a specific downstream task, the model achieves high accuracy in predicting the sentiment of movie reviews.
- Downloads last month
- 1
Model tree for PardisSzah/CS546_HW1
Base model
sentence-transformers/all-MiniLM-L6-v2