Views
No views yet
1from transformers import AutoConfig, AutoModelForSequenceClassification, AutoTokenizer
2import torch
3
4
5config = AutoConfig.from_pretrained("erythropygia/distilbert-turkish-sentiment-analysis")
6tokenizer = AutoTokenizer.from_pretrained("erythropygia/distilbert-turkish-sentiment-analysis", config=config)
7model = AutoModelForSequenceClassification.from_pretrained("erythropygia/distilbert-turkish-sentiment-analysis", config=config)
8
9import torch
10def evaluate(text):
11 inputs = tokenizer(text, return_tensors="pt")
12
13 with torch.no_grad():
14 outputs = model(**inputs)
15
16 logits = outputs.logits
17 softmax_logits = torch.softmax(logits, dim=1)
18 max_prob, predicted = torch.max(softmax_logits, dim=1)
19
20 predicted_label = config.id2label[predicted.item()]
21 predicted_probability = max_prob.item()
22 return predicted_label, predicted_probability
23
24model_data = ["Bu ürünü çok beğendim mağazayı tavsiye ederim.",
25 "Bugün çok üzgünüm.", "Kendimi öldüreceğim.",
26 "Çok memnunum, kesinlikle herkese tavsiye ederim. ayrıca hepsiburada ya süper hızlı gönderim için teşekkür ederim.",
27 "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.",
28 "izlediğim vakti boşa hacadığımı düşünüorum, o kadar kötüydü malesef...",]
29
30results = {}
31
32for text in model_data:
33 predicted_label, predicted_probability = evaluate(text.lower())
34 results[text] = {"label": predicted_label, "probability": predicted_probability}
35
36results
37
38'''
39{'Bu ürünü çok beğendim mağazayı tavsiye ederim.': {'label': 'positive',
40 'probability': 0.9934958219528198},
41 'Bugün çok üzgünüm.': {'label': 'negative',
42 'probability': 0.8088045120239258},
43 'Kendimi öldüreceğim.': {'label': 'negative',
44 'probability': 0.9505906701087952},
45 'Çok memnunum, kesinlikle herkese tavsiye ederim. ayrıca hepsiburada ya süper hızlı gönderim için teşekkür ederim.': {'label': 'positive',
46 'probability': 0.9969924688339233},
47 '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',
48 'probability': 0.7242251634597778},
49 'izlediğim vakti boşa hacadığımı düşünüorum, o kadar kötüydü malesef...': {'label': 'negative',
50 'probability': 0.9954615235328674}}
51'''