Upload README.md

README.md

@@ -1,63 +1,80 @@
----
-language: german
-widget:
-- text: "It has been determined that the amount of greenhouse gases have decreased by almost half because of the prevalence in the utilization of nuclear power."
----
 
 ### Welcome to ParlBERT-Topic-German!
 
-🤖 **Model description**
+🏷 **Model description**
 
-This model was trained on ~10k manually annotated political requests (📚 [Stab et al. 2018](https://www.aclweb.org/anthology/D18-1402/)) of comparative agenda topics to classify text into one of twenty labels: 🏷 **TOPIC1** (0) and **TOPIC2** (1) ...
+This model was trained on \~10k manually annotated political interpellations (📚 [Breunig/ Schnatterer 2019](https://oxford.universitypressscholarship.com/view/10.1093/oso/9780198835332.001.0001/oso-9780198835332)) of comparative agenda topics to classify text into one of twenty labels (annotation codebook).
 
 🗃 **Dataset**
 
-The dataset (📚 Stab et al. 2018) consists of **ARGUMENTS** (\~11k) that either support or oppose a topic if it includes a relevant reason for supporting or opposing the topic, or as a **NON-ARGUMENT** (\~14k) if it does not include reasons. The authors focus on controversial topics, i.e., topics that include "an obvious polarity to the possible outcomes" and compile a final set of eight controversial topics: _abortion, school uniforms, death penalty, marijuana legalization, nuclear energy, cloning, gun control, and minimum wage_.
-
-| TOPIC | ARGUMENT | NON-ARGUMENT |
+| party | speeches | tokens |
 |----|----|----|
-| abortion | 2213 | 2,427 |
-| school uniforms | 325 | 1,734 |
-| death penalty | 325 | 2,083 |
-| marijuana legalization | 325 | 1,262 |
-| nuclear energy | 325 | 2,118 |
-| cloning | 325 | 1,494 |
-| gun control | 325 | 1,889 |
-| minimum wage | 325 | 1,346 |
+| CDU/CSU | 7,635 | 4,862,654 |
+| SPD | 5,321 | 3,158,315 |
+| AfD | 3,465 | 1,844,707 |
+| FDP | 3,067 | 1,593,108 |
+| The Greens | 2,866 | 1,522,305 |
+| The Left | 2,671 | 1,394,089 |
+| cross-bencher | 200 | 86,170 |
 
 🏃🏼‍♂️**Model training**
 
-**ParlBERT-Topic** was fine-tuned on ParlBERT from HuggingFace for topic modeling with questions dataset from the Comparative Agendas Project. We used the HuggingFace trainer with the following hyperparameters:
+**ParlBERT-Topic** was fine-tuned on a domain adapted model for topic modeling with interpellations dataset from the Comparative Agendas Project (mlm\_probability=.15). We used the HuggingFace trainer with the following hyperparameters.
+
+🤖 ** Use ** 
 
 ```
-training_args = TrainingArguments(
-    num_train_epochs=2,
-    learning_rate=2.3102e-06,
-    seed=8,
-    per_device_train_batch_size=64,
-    per_device_eval_batch_size=64,
-)
+from transformers import pipeline
+
+pipeline_classification_topics = pipeline("text-classification", model="chkla/parlbert-topics-german", tokenizer="bert-base-german-cased", return_all_scores=False, device=0)
+
+text = "Sachgebiet Ausschließliche Gesetzgebungskompetenz des Bundes über die Zusammenarbeit des Bundes und der Länder zum Schutze der freiheitlichen demokratischen Grundordnung, des Bestandes und der Sicherheit des Bundes oder eines Landes Wir fragen die Bundesregierung"
+
+pipeline_classification_topics(text) # Government
+
 ```
 
+
 📊 **Evaluation**
 
 The model was evaluated on an evaluation set (20%):
 
-| Model | Acc | F1 | R arg | R non | P arg | P non |
-|----|----|----|----|----|----|----|
-| RoBERTArg | 0.8193 | 0.8021 | 0.8463 | 0.7986 | 0.7623 | 0.8719 |
-
-Showing the **confusion matrix** using again the evaluation set:
-
-| | ARGUMENT | NON-ARGUMENT |
+| Label | F1 | support |
 |----|----|----|
-| ARGUMENT | 2213 | 558 |
-| NON-ARGUMENT | 325 | 1790 |
+| International | 80.0 | 1,126 |
+| Defense | 85.0 | 1,099 |
+| Government | 71.3 | 989 |
+| International | 76.5 | 978 |
+| International | 76.6 | 845 |
+| International | 86.0 | 800 |
+| International | 67.1 | 0.8021 |
+| International | 78.6 | 0.8021 |
+| International | 78.2 | 0.8021 |
+| International | 64.4 | 0.8021 |
+| International | 81.0 | 0.8021 |
+| International | 69.1 | 0.8021 |
+| International | 62.8 | 0.8021 |
+| International | 76.3 | 0.8021 |
+| International | 49.2 | 0.8021 |
+| International | 63.0 | 0.8021 |
+| International | 71.6 | 0.8021 |
+| International | 79.6 | 0.8021 |
+| International | 61.5 | 0.8021 |
+| International | 45.4 | 0.8021 |
+
 
 ⚠️ **Intended Uses & Potential Limitations**
 
 The model can only be a starting point to dive into the exciting field of policy topic classification in political texts. But be aware. Models are often highly topic dependent. Therefore, the model may perform less well on different topics and text types not included in the training set.
 
-Enjoy and stay tuned! 🚀
+👥 ** Cite **
+```
+@article{klamm2022frameast,
+  title={FrameASt: A Framework for Second-level Agenda Setting in Parliamentary Debates through the Lense of Comparative Agenda Topics},
+  author={Klamm, Christopher and Rehbein, Ines and Ponzetto, Simone},
+  journal={ParlaCLARIN III at LREC2022},
+  year={2022}
+}
+```
 
 🐦 Twitter: [@chklamm](http://twitter.com/chklamm)