Views
No views yet
| Index | Dimension |
|---|---|
| 0 | Anti-Elitism |
| 1 | People-Centrism |
| 2 | Left-Wing Host-Ideology |
| 3 | Right-Wing Host-Ideology |
1import torch
2from transformers import AutoModelForSequenceClassification
3from transformers import AutoTokenizer
4
5# load tokenizer
6tokenizer = AutoTokenizer.from_pretrained("luerhard/PopBERT")
7
8# load model
9model = AutoModelForSequenceClassification.from_pretrained("luerhard/PopBERT")
10
11# define text to be predicted
12text = (
13 "Das ist Klassenkampf von oben, das ist Klassenkampf im Interesse von "
14 "Vermögenden und Besitzenden gegen die Mehrheit der Steuerzahlerinnen und "
15 "Steuerzahler auf dieser Erde."
16)
17
18# encode text with tokenizer
19encodings = tokenizer(text, return_tensors="pt")
20
21# predict
22with torch.inference_mode():
23 out = model(**encodings)
24
25# get probabilties
26probs = torch.nn.functional.sigmoid(out.logits)
27print(probs.detach().numpy())[[0.8765146 0.34838045 0.983123 0.02148379]][0.415961, 0.295400, 0.429109, 0.302714]| Dimension | Precision | Recall | F1 |
|---|---|---|---|
| Anti-Elitism | 0.81 | 0.88 | 0.84 |
| People-Centrism | 0.70 | 0.73 | 0.71 |
| Left-Wing Ideology | 0.69 | 0.77 | 0.73 |
| Right-Wing Ideology | 0.68 | 0.66 | 0.67 |
| --- | --- | --- | --- |
| micro avg | 0.75 | 0.80 | 0.77 |
| macro avg | 0.72 | 0.76 | 0.74 |