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
Safetensors
Model size
22.7M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for PardisSzah/CS546_HW1

Finetuned
(719)
this model

Dataset used to train PardisSzah/CS546_HW1