Views
No views yet
bert-base-multilingual-cased. The model was trained using the Georgian Sentiment Analysis dataset.bert-base-multilingual-casedArseniy-Sandalov/Georgian-Sentiment-Analysis1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2import torch
3
4model_name = "Arseniy-Sandalov/GeorgianBert-Sent"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
8def predict_sentiment(text):
9 inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
10 with torch.no_grad():
11 outputs = model(**inputs)
12 prediction = torch.argmax(outputs.logits, dim=1).item()
13 return ["negative", "neutral", "positive"][prediction]
14
15text = "ახალი მეარი კარგია ერთილა"
16print(predict_sentiment(text))@misc {Stefanovitch2023Sentiment,
author = {Stefanovitch, Nicolas and Piskorski, Jakub and Kharazi, Sopho},
title = {Sentiment analysis for Georgian},
year = {2023},
publisher = {European Commission, Joint Research Centre (JRC)},
howpublished = {\url{http://data.europa.eu/89h/9f04066a-8cc0-4669-99b4-f1f0627fdbbf}},
url = {http://data.europa.eu/89h/9f04066a-8cc0-4669-99b4-f1f0627fdbbf},
type = {dataset},
note = {PID: http://data.europa.eu/89h/9f04066a-8cc0-4669-99b4-f1f0627fdbbf}
}