billerjully/ru-absa-restaurant-reviews
Viewer • Updated • 7.25k • 59
How to use billerjully/BERT-absa-rest-reviews-asc with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="billerjully/BERT-absa-rest-reviews-asc") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("billerjully/BERT-absa-rest-reviews-asc")
model = AutoModelForSequenceClassification.from_pretrained("billerjully/BERT-absa-rest-reviews-asc", device_map="auto")Document-level sentiment classifier for Russian restaurant reviews. Given the full text of a review, it predicts the overall sentiment: Positive, Negative, or Neutral.
This is a fine-tune of cointegrated/rubert-tiny2 (a compact Russian BERT, 29M parameters) for sequence classification. It complements the aspect extraction model billerjully/BERT-absa-rest-reviews-ate: together they answer "what is the review about" and "how does the reviewer feel overall".
from transformers import pipeline
asc = pipeline(
"text-classification",
model="billerjully/BERT-absa-rest-reviews-asc",
top_k=None, # return probabilities for all three classes
)
text = "Ужасное место. Еда пришла холодной, официант хамил, цены завышены."
print(asc(text)[0])
[{'label': 'Positive', 'score': 0.71},
{'label': 'Negative', 'score': 0.15},
{'label': 'Neutral', 'score': 0.14}]
Trained on billerjully/ru-absa-restaurant-reviews — 7,253 Russian restaurant reviews:
| Label | Count | Share |
|---|---|---|
| Positive | 6,163 | 85.0% |
| Negative | 591 | 8.1% |
| Neutral | 499 | 6.9% |
Base model
cointegrated/rubert-tiny2