Views
No views yet
aubmindlab/bert-base-arabertv02-twitter| ID | Label | Meaning |
|---|---|---|
| 0 | Favor | Tweet supports the target |
| 1 | Against | Tweet opposes the target |
| 2 | None | Neutral, irrelevant, or unclear stance |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("HassanB4/t1_s3_arabert_twitter_text_target")
5model = AutoModelForSequenceClassification.from_pretrained("HassanB4/t1_s3_arabert_twitter_text_target")
6model.eval()
7
8id2label = {0: "Favor", 1: "Against", 2: "None"}
9
10text = "..."
11target = "..."
12inputs = tokenizer(text, target, return_tensors="pt", truncation=True, max_length=512)
13
14with torch.no_grad():
15 logits = model(**inputs).logits
16
17predicted_label = id2label[int(torch.argmax(logits, dim=-1)[0])]
18print(predicted_label)