Downloads · 30 days
6
1% of all-time downloads
victorwkey/tourism-categories-bert
tourism-categories-bert is a text classification model from victorwkey. Use it when you need a label for a piece of text. The card lists the license as mit.
A BERT-based model fine-tuned for multi-label classification of tourism reviews into categories.
Downloads · 30 days
6
1% of all-time downloads
All-time downloads
485
Public
Parameters
167M
669 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors669 MB · 99%
From the Hugging Face model README
A BERT-based model fine-tuned for multi-label classification of tourism reviews into categories.
This model was trained as part of the AI Tourism Opinion Analyzer project to classify customer reviews into multiple categories like: location, service, cleanliness, etc.
from transformers import AutoModelForSequenceClassification, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("victorwkey/tourism-categories-bert")
model = AutoModelForSequenceClassification.from_pretrained("victorwkey/tourism-categories-bert")
# Example
text = "La ubicación del hotel es excelente y el servicio fue muy amable"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)