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