Downloads · 30 days
6
1% of all-time downloads
minilmFT_TripAdvisor_Sentiment
gosorio/minilmFT_TripAdvisor_Sentiment
minilmFT_TripAdvisor_Sentiment is a text classification model from gosorio. Use it when you need a label for a piece of text. It is set up for transformers.
Sentiment analysis model that uses MiniLM pre-trained (from https://huggingface.co/microsoft/MiniLM-L12-H384-uncased), and fine-tuned on a dataset containing Trip Advisor reviews (from https://www.kaggle.com/datasets/…
All-time downloads
989
Public
Repo size
267 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin134 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jul 9, 2023
- Updated
- Jul 9, 2023
- SHA
- 16763042
Try a prompt
Model card
From the Hugging Face model README
Sentiment analysis model that uses MiniLM pre-trained (from https://huggingface.co/microsoft/MiniLM-L12-H384-uncased), and fine-tuned on a dataset containing Trip Advisor reviews (from https://www.kaggle.com/datasets/arnabchaki/tripadvisor-reviews-2023).
Reviews with 1 or 2 stars are considered 'Negative', 3 stars are 'Neutral', and 4 or 5 stars are 'Positive'.
Should be loaded with the following code:
# Load pre-trained model and tokenizer
model_name = "gosorio/minilmFT_TripAdvisor_Sentiment"
tokenizer_name = "microsoft/MiniLM-L12-H384-uncased" # the standard MiniLM
device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3).to(device)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Jul 9, 2023
- Updated
- Jul 9, 2023