Update README.md
Browse files
README.md
CHANGED
@@ -55,7 +55,7 @@ probabilities = torch.softmax(logits, dim=1).tolist()[0]
|
|
55 |
probabilities = {model.config.id2label[index]: round(probability * 100, 2) for index, probability in enumerate(probabilities)}
|
56 |
probabilities = dict(sorted(probabilities.items(), key=lambda item: item[1], reverse=True))
|
57 |
print(probabilities)
|
58 |
-
# {'201 - Freedom and Human Rights':
|
59 |
|
60 |
predicted_class = model.config.id2label[logits.argmax().item()]
|
61 |
print(predicted_class)
|
@@ -103,7 +103,7 @@ The model was evaluated on a test set of 200,920 annotated manifesto statements.
|
|
103 |
|
104 |
Please cite the model as follows:
|
105 |
|
106 |
-
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.context.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.context.2024.1.1
|
107 |
|
108 |
```bib
|
109 |
@misc{Burst:2024,
|
@@ -112,6 +112,6 @@ Burst, Tobias / Lehmann, Pola / Franzmann, Simon / Al-Gaddooa, Denise / Ivanusch
|
|
112 |
Publisher = {Wissenschaftszentrum Berlin für Sozialforschung / Göttinger Institut für Demokratieforschung},
|
113 |
Title = {manifestoberta. Version 56topics.context.2024.1.1},
|
114 |
doi = {10.25522/manifesto.manifestoberta.56topics.context.2024.1.1},
|
115 |
-
url = {https://doi.org/10.25522/manifesto.manifestoberta.56topics.context.2024.1.1},
|
116 |
Year = {2024},
|
117 |
```
|
|
|
55 |
probabilities = {model.config.id2label[index]: round(probability * 100, 2) for index, probability in enumerate(probabilities)}
|
56 |
probabilities = dict(sorted(probabilities.items(), key=lambda item: item[1], reverse=True))
|
57 |
print(probabilities)
|
58 |
+
# {'201 - Freedom and Human Rights': 86.58, '107 - Internationalism: Positive': 8.12, '106 - Peace': 1.67...
|
59 |
|
60 |
predicted_class = model.config.id2label[logits.argmax().item()]
|
61 |
print(predicted_class)
|
|
|
103 |
|
104 |
Please cite the model as follows:
|
105 |
|
106 |
+
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.context.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.context.2024.1.1
|
107 |
|
108 |
```bib
|
109 |
@misc{Burst:2024,
|
|
|
112 |
Publisher = {Wissenschaftszentrum Berlin für Sozialforschung / Göttinger Institut für Demokratieforschung},
|
113 |
Title = {manifestoberta. Version 56topics.context.2024.1.1},
|
114 |
doi = {10.25522/manifesto.manifestoberta.56topics.context.2024.1.1},
|
115 |
+
url = {https://doi.org/10.25522/manifesto.manifestoberta.56topics.context.2024.1.1},
|
116 |
Year = {2024},
|
117 |
```
|