Views
No views yet
1import numpy as np
2from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
3#loading model
4tokenizer = AutoTokenizer.from_pretrained("nimaafshar/parsbert-fa-sentiment-twitter")
5
6model = TFAutoModelForSequenceClassification.from_pretrained("nimaafshar/parsbert-fa-sentiment-twitter")
7classes = ["negative","neutral","positive"]1
2#using model
3sequences = [".غذا خیلی افتضاح بود متاسفم برای مدیریت رستورن خیلی بد بود.",
4 "خیلی خوشمزده و عالی بود عالی",
5"میتونم اسمتونو بپرسم؟"
6]
7
8for sequence in sequences:
9 inputs = tokenizer(sequence, return_tensors="tf")
10 classification_logits = model(inputs)[0]
11 results = tf.nn.softmax(classification_logits, axis=1).numpy()[0]
12 print(classes[np.argmax(results)])
13 percentages = np.around(results*100)
14 print(percentages)