erythropygia/distilbert-turkish-sentiment-analysis

This model performs sentiment detect on Turkish texts. It gives sevent outputs:

• "0": "negative",
• "1": "positive"

from transformers import AutoConfig, AutoModelForSequenceClassification, AutoTokenizer
import torch


config = AutoConfig.from_pretrained("erythropygia/distilbert-turkish-sentiment-analysis")
tokenizer = AutoTokenizer.from_pretrained("erythropygia/distilbert-turkish-sentiment-analysis", config=config)
model = AutoModelForSequenceClassification.from_pretrained("erythropygia/distilbert-turkish-sentiment-analysis", config=config)

import torch
def evaluate(text):
    inputs = tokenizer(text, return_tensors="pt")

    with torch.no_grad():
        outputs = model(**inputs)

    logits = outputs.logits
    softmax_logits = torch.softmax(logits, dim=1)
    max_prob, predicted = torch.max(softmax_logits, dim=1)

    predicted_label = config.id2label[predicted.item()]
    predicted_probability = max_prob.item()
    return predicted_label, predicted_probability

model_data = ["Bu ürünü çok beğendim mağazayı tavsiye ederim.",
              "Bugün çok üzgünüm.", "Kendimi öldüreceğim.",
              "Çok memnunum, kesinlikle herkese tavsiye ederim. ayrıca hepsiburada ya süper hızlı gönderim için teşekkür ederim.",
              "2 yıl içerisinde 3 kez servise götürdüm ve hala aynı sorunları devam ediyor. Başka bir ütü deneyin yoksa pişman olursunuz.",
              "izlediğim vakti boşa hacadığımı düşünüorum, o kadar kötüydü malesef...",]

results = {}

for text in model_data:
    predicted_label, predicted_probability = evaluate(text.lower())
    results[text] = {"label": predicted_label, "probability": predicted_probability}

results

'''
{'Bu ürünü çok beğendim mağazayı tavsiye ederim.': {'label': 'positive',
  'probability': 0.9934958219528198},
 'Bugün çok üzgünüm.': {'label': 'negative',
  'probability': 0.8088045120239258},
 'Kendimi öldüreceğim.': {'label': 'negative',
  'probability': 0.9505906701087952},
 'Çok memnunum, kesinlikle herkese tavsiye ederim. ayrıca hepsiburada ya süper hızlı gönderim için teşekkür ederim.': {'label': 'positive',
  'probability': 0.9969924688339233},
 '2 yıl içerisinde 3 kez servise götürdüm ve hala aynı sorunları devam ediyor. Başka bir ütü deneyin yoksa pişman olursunuz.': {'label': 'negative',
  'probability': 0.7242251634597778},
 'izlediğim vakti boşa hacadığımı düşünüorum, o kadar kötüydü malesef...': {'label': 'negative',
  'probability': 0.9954615235328674}}
'''