# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Tobias/bert-base-uncased_English_Hotel_sentiment")
model = AutoModelForSequenceClassification.from_pretrained("Tobias/bert-base-uncased_English_Hotel_sentiment")Quick Links
English Hotel Review Sentiment Classification
A model trained on English Hotel Reviews from Switzerland. The base model is the bert-base-uncased. The last hidden layer of the base model was extracted and a classification layer was added. The entire model was then trained for 5 epochs on our dataset.
- Downloads last month
- 11
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Tobias/bert-base-uncased_English_Hotel_sentiment")