Update README.md
Browse files
README.md
CHANGED
@@ -13,15 +13,15 @@ pinned: false
|
|
13 |
|
14 |
## Model
|
15 |
|
16 |
-
* Fine tuned model from SieBERT checkpoint (RoBERTa large). See
|
17 |
* Binary Classification
|
18 |
* Dataset: 7,958 positive sentiments (label 1) and 1,437 negative sentiments (label 0)
|
19 |
* Train / Test : 80/20
|
20 |
* LEARNING_RATE = 1e-05
|
21 |
* 3 Epochs
|
22 |
* Preclassifier 1024*1024 with dropout 0.1
|
23 |
-
* Test data: F-measure label
|
24 |
-
* Test data: F-measure label
|
25 |
|
26 |
|
27 |
|
|
|
13 |
|
14 |
## Model
|
15 |
|
16 |
+
* Fine tuned model from SieBERT checkpoint (RoBERTa large). See [SieBERT](https://huggingface.co/siebert/sentiment-roberta-large-english)
|
17 |
* Binary Classification
|
18 |
* Dataset: 7,958 positive sentiments (label 1) and 1,437 negative sentiments (label 0)
|
19 |
* Train / Test : 80/20
|
20 |
* LEARNING_RATE = 1e-05
|
21 |
* 3 Epochs
|
22 |
* Preclassifier 1024*1024 with dropout 0.1
|
23 |
+
* Test data: F-measure label 1: 0.9736
|
24 |
+
* Test data: F-measure label 0: 0.8547
|
25 |
|
26 |
|
27 |
|