danyaljj commited on
Commit
e2585a3
2 Parent(s): 34c54d8 58ba977

Merge branch 'main' of https://huggingface.co/persiannlp/mt5-base-parsinlu-sentiment-analysis into main

Browse files
Files changed (1) hide show
  1. README.md +82 -0
README.md ADDED
@@ -0,0 +1,82 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - fa
4
+ - multilingual
5
+ thumbnail: "https://upload.wikimedia.org/wikipedia/commons/a/a2/Farsi.svg"
6
+ tags:
7
+ - sentiment
8
+ - sentiment-analysis
9
+ - mt5
10
+ - persian
11
+ - farsi
12
+ license: "CC BY-NC-SA 4.0"
13
+ datasets:
14
+ - parsinlu
15
+ metrics:
16
+ - accuracy
17
+ ---
18
+
19
+ # Sentiment Analysis (آنالیز احساسات)
20
+
21
+ This is a mT5 model for sentiment analysis.
22
+ Here is an example of how you can run this model:
23
+
24
+ ```python
25
+ import torch
26
+ from transformers import MT5ForConditionalGeneration, MT5Tokenizer
27
+ import numpy as np
28
+
29
+ model_name_or_path = "persiannlp/mt5-base-parsinlu-sentiment-analysis"
30
+ tokenizer = MT5Tokenizer.from_pretrained(model_name)
31
+ model = MT5ForConditionalGeneration.from_pretrained(model_name)
32
+
33
+
34
+ def model_predict(text_a, text_b):
35
+ features = tokenizer( [(text_a, text_b)], padding="max_length", truncation=True, return_tensors='pt')
36
+ output = model(**features)
37
+ logits = output[0]
38
+ probs = torch.nn.functional.softmax(logits, dim=1).tolist()
39
+ idx = np.argmax(np.array(probs))
40
+ print(labels[idx], probs)
41
+
42
+
43
+ def run_model(context, query, **generator_args):
44
+ input_ids = tokenizer.encode(context + "<sep>" + query, return_tensors="pt")
45
+ res = model.generate(input_ids, **generator_args)
46
+ output = tokenizer.batch_decode(res, skip_special_tokens=True)
47
+ print(output)
48
+ return output
49
+
50
+
51
+ run_model(
52
+ "یک فیلم ضعیف بی محتوا بدون فیلمنامه . شوخی های سخیف .",
53
+ "نظر شما در مورد داستان، فیلمنامه، دیالوگ ها و موضوع فیلم لونه زنبور چیست؟"
54
+ )
55
+
56
+ run_model(
57
+ "فیلم تا وسط فیلم یعنی دقیقا تا جایی که معلوم میشه بچه های املشی دنبال رضان خیلی خوب و جذاب پیش میره ولی دقیقا از همونجاش سکته میزنه و خلاص...",
58
+ "نظر شما به صورت کلی در مورد فیلم ژن خوک چیست؟"
59
+ )
60
+ run_model(
61
+ "اصلا به هیچ عنوان علاقه نداشتم اجرای می سی سی پی نشسته میمیرد روی پرده سینما ببینم دیالوگ های تکراری هلیکوپتر ماشین آلندلون لئون پاپیون آخه چرااااااااااااااا همون حسی که توی تالار وحدت بعد از نیم ساعت به سرم اومد امشب توی سالن سینما تجربه کردم ،حس گریز از سالن.......⁦ ⁦(ノಠ益ಠ)ノ⁩ ",
62
+ " نظر شما در مورد صداگذاری و جلوه های صوتی فیلم مسخره‌باز چیست؟"
63
+ )
64
+
65
+ run_model(
66
+ " گول نخورید این رنگارنگ مینو نیست برای شرکت گرجیه و متاسفانه این محصولش اصلا مزه رنگارنگی که انتظار دارید رو نمیده ",
67
+ " نظر شما در مورد عطر، بو، و طعم این بیسکویت و ویفر چیست؟"
68
+ )
69
+
70
+ run_model(
71
+ "در مقایسه با سایر برندهای موجود در بازار با توجه به حراجی که داشت ارزانتر ب",
72
+ " شما در مورد قیمت و ارزش خرید این حبوبات و سویا چیست؟"
73
+ )
74
+
75
+ run_model(
76
+ "من پسرم عاشق ایناس ولی دیگه به خاطر حفظ محیط زیست فقط زمانهایی که مجبور باشم شیر دونه ای میخرم و سعی میکنم دیگه کمتر شیر با بسته بندی تتراپک استفاده کنم ",
77
+ "نظر شما به صورت کلی در مورد این شیر چیست؟"
78
+ )
79
+ ```
80
+
81
+
82
+ For more details, visit this page: https://github.com/persiannlp/parsinlu/