Tourism Opinion Category Classifier
A BERT-based model fine-tuned for multi-label classification of tourism reviews into categories.
Model Description
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.
Usage
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)
Training
- Base model: BERT multilingual
- Task: Multi-label classification
- Dataset: Custom tourism reviews dataset
- Categories: Location, Service, Cleanliness, Value, Amenities, etc.
- Downloads last month
- 31