Update README.md
Browse files
README.md
CHANGED
@@ -46,7 +46,7 @@ probabilities = torch.softmax(logits, dim=1).tolist()[0]
|
|
46 |
probabilities = {model.config.id2label[index]: round(probability * 100, 2) for index, probability in enumerate(probabilities)}
|
47 |
probabilities = dict(sorted(probabilities.items(), key=lambda item: item[1], reverse=True))
|
48 |
print(probabilities)
|
49 |
-
# {'501 - Environmental Protection: Positive': 67.
|
50 |
|
51 |
predicted_class = model.config.id2label[logits.argmax().item()]
|
52 |
print(predicted_class)
|
@@ -69,7 +69,7 @@ The model was evaluated on a test set of 200,920 annotated manifesto statements.
|
|
69 |
|
70 |
Please cite the model as follows:
|
71 |
|
72 |
-
Burst, Tobias / Lehmann, Pola / Franzmann, Simon / Al-Gaddooa, Denise / Ivanusch, Christoph / Regel, Sven / Riethmüller, Felicia / Weßels, Bernhard / Zehnter, Lisa (2024): manifestoberta. Version 56topics.sentence.2024.1.1. Berlin: Wissenschaftszentrum Berlin für Sozialforschung (WZB) / Göttingen: Institut für Demokratieforschung (IfDem). https://doi.org/10.25522/manifesto.manifestoberta.56topics.sentence.2024.1.1
|
73 |
|
74 |
```bib
|
75 |
@misc{Burst:2024,
|
@@ -78,6 +78,6 @@ Burst, Tobias / Lehmann, Pola / Franzmann, Simon / Al-Gaddooa, Denise / Ivanusch
|
|
78 |
Publisher = {Wissenschaftszentrum Berlin für Sozialforschung / Göttinger Institut für Demokratieforschung},
|
79 |
Title = {manifestoberta. Version 56topics.sentence.2024.1.1},
|
80 |
doi = {10.25522/manifesto.manifestoberta.56topics.sentence.2024.1.1},
|
81 |
-
url = {https://doi.org/10.25522/manifesto.manifestoberta.56topics.sentence.2024.1.1},
|
82 |
Year = {2024},
|
83 |
```
|
|
|
46 |
probabilities = {model.config.id2label[index]: round(probability * 100, 2) for index, probability in enumerate(probabilities)}
|
47 |
probabilities = dict(sorted(probabilities.items(), key=lambda item: item[1], reverse=True))
|
48 |
print(probabilities)
|
49 |
+
# {'501 - Environmental Protection: Positive': 67.56, '411 - Technology and Infrastructure': 14.03, '107 - Internationalism: Positive': 13.58, '416 - Anti-Growth Economy: Positive': 2.24...
|
50 |
|
51 |
predicted_class = model.config.id2label[logits.argmax().item()]
|
52 |
print(predicted_class)
|
|
|
69 |
|
70 |
Please cite the model as follows:
|
71 |
|
72 |
+
Burst, Tobias / Lehmann, Pola / Franzmann, Simon / Al-Gaddooa, Denise / Ivanusch, Christoph / Regel, Sven / Riethmüller, Felicia / Weßels, Bernhard / Zehnter, Lisa (2024): manifestoberta. Version 56topics.sentence.2024.1.1. Berlin: Wissenschaftszentrum Berlin für Sozialforschung (WZB) / Göttingen: Institut für Demokratieforschung (IfDem). https://doi.org/10.25522/manifesto.manifestoberta.56topics.sentence.2024.1.1
|
73 |
|
74 |
```bib
|
75 |
@misc{Burst:2024,
|
|
|
78 |
Publisher = {Wissenschaftszentrum Berlin für Sozialforschung / Göttinger Institut für Demokratieforschung},
|
79 |
Title = {manifestoberta. Version 56topics.sentence.2024.1.1},
|
80 |
doi = {10.25522/manifesto.manifestoberta.56topics.sentence.2024.1.1},
|
81 |
+
url = {https://doi.org/10.25522/manifesto.manifestoberta.56topics.sentence.2024.1.1},
|
82 |
Year = {2024},
|
83 |
```
|