Hugging Face's logo

chkla
/
roberta-argument

Text Classification
Transformers
PyTorch
TensorFlow
JAX
Safetensors
English
roberta
Inference Endpoints
Model card Files Community
3
chkla commited on
Commit
d0a1aa1
•
1 Parent(s): 853149d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -3
README.md CHANGED
@@ -1,12 +1,19 @@
1
  Welcome to **RoBERTArg**!
2
 
3
- 🤖 **Training**: Model (RoBERTA_base_) fine-tuned on a heterogeneous argument corpus (~40K sentences, Stab et al. (2018) 📚) corpus of several controversial topics (abortion etc.).
4
 
5
- 🏷 Using **NON-ARGUMENT** (0) and **ARGUMENT** (1) as labels.
6
 
 
7
  | Model | Acc | F1 | R arg | R non | P arg | P non |
8
  |----|----|----|----|----|----|----|
9
- | robertArg | 0.7881 | 0.7989 | 0.8147 | 0.7557 | 0.8496 |
 
 
 
 
 
 
10
 
11
  👉🏾 Check out _chkla/argument-analyzer/_ for more details.
12
 
 
1
  Welcome to **RoBERTArg**!
2
 
3
+ 🤖 ** Training **: Model (RoBERTA_base_) fine-tuned on a heterogeneous argument corpus (~40K sentences, Stab et al. (2018) 📚) corpus of several controversial topics (abortion etc.).
4
 
5
+ 🏷 Using ** NON-ARGUMENT ** (0) and ** ARGUMENT ** (1) as labels.
6
 
7
+ 📊 ** Performance **
8
  | Model | Acc | F1 | R arg | R non | P arg | P non |
9
  |----|----|----|----|----|----|----|
10
+ | robertArg | 0.80577 | 0.78407| 0.8147 | 0.7989 | 0.7557 | 0.8496 |
11
+
12
+ ** Confusion Matrix **
13
+ | | ** 1 ** | ** 0 ** |
14
+ |----|----|----|
15
+ | ** 1 ** | 2214 | 557 |
16
+ | ** 0 ** | 392 | 1723 |
17
 
18
  👉🏾 Check out _chkla/argument-analyzer/_ for more details.
19