Update README.md

README.md

@@ -2,11 +2,22 @@ Welcome to **RoBERTArg**!
 
 🤖 **Model description**
 
-This model was trained on ~40k heterogeneous manually annotated sentences (📚 Stab et al. 2018) of controversial topics to classify text into one of two labels: 🏷 **NON-ARGUMENT** (0) and **ARGUMENT** (1).
+This model was trained on ~25k heterogeneous manually annotated sentences (📚 Stab et al. 2018) of controversial topics to classify text into one of two labels: 🏷 **NON-ARGUMENT** (0) and **ARGUMENT** (1).
 
 **Dataset**
 
-The dataset (📚 Stab et al. 2018) consists of **ARGUMENTS** that either support or oppose a topic if it includes a relevant reason for supporting or opposing the topic, or as a **NON-ARGUMENT** 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_.
+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 |
+|----|----|----|
+| 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 |
 
 **Model training**