Update README.md
Browse files
README.md
CHANGED
@@ -3,7 +3,7 @@ tags:
|
|
3 |
- generated_from_trainer
|
4 |
model-index:
|
5 |
- name: drug-stance-bert
|
6 |
-
results: []
|
7 |
---
|
8 |
|
9 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -11,33 +11,42 @@ should probably proofread and complete it, then remove this comment. -->
|
|
11 |
|
12 |
# drug-stance-bert
|
13 |
|
14 |
-
This model
|
15 |
|
16 |
## Model description
|
17 |
|
18 |
-
|
19 |
|
20 |
-
## Intended uses & limitations
|
21 |
|
22 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
|
24 |
-
|
|
|
|
|
|
|
25 |
|
26 |
-
|
|
|
|
|
27 |
|
28 |
## Training procedure
|
29 |
-
|
30 |
### Training hyperparameters
|
31 |
|
32 |
The following hyperparameters were used during training:
|
33 |
- learning_rate: 5e-05
|
34 |
-
- train_batch_size:
|
35 |
-
- eval_batch_size:
|
36 |
- seed: 42
|
37 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
38 |
- lr_scheduler_type: linear
|
39 |
-
-
|
40 |
-
- num_epochs: 4
|
41 |
|
42 |
### Framework versions
|
43 |
|
|
|
3 |
- generated_from_trainer
|
4 |
model-index:
|
5 |
- name: drug-stance-bert
|
6 |
+
results: [1, 0, -1]
|
7 |
---
|
8 |
|
9 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
11 |
|
12 |
# drug-stance-bert
|
13 |
|
14 |
+
This model is a fine-tuned version of [cardiffnlp/twitter-roberta-base-sentiment](https://huggingface.co/cardiffnlp/twitter-roberta-base-sentiment) on [COVID-CQ](https://github.com/eceveco/COVID-CQ), a dataset that contains 3-label annotated opinions (negative, neutral, and positive) of the tweet initiators regarding the use of Chloroquine or Hydroxychloroquine for the treatment or prevention of the coronavirus.
|
15 |
|
16 |
## Model description
|
17 |
|
18 |
+
"We developed two COVID-drug-stance RoBERTa-base models by fine-tuning a pre-trained Twitter-specific stance detection model on a stance data set called COVID-CQ. The data were divided into training-dev-test validation datasets with a 70:10:20 ratio. Model I (COVID-drug-stance-BERT) was trained on the original tweet data, and Model II (COVID-drug-stance-BERT-masked) was trained on tweets with drug names masked as “[mask]” for model generalizability on different drugs. The two models had similar performance on the COVID-19 validation set: COVID-drug-stance-BERT had an accuracy of 86.88%, and the masked model had an accuracy of 86.67%. The two models were then evaluated by predicting tweet initiators’ attitudes towards the drug mentioned in each tweet using randomly selected test sets (100 tweets) of each drug (Hydroxychloquine, Ivermectin, Molnupiravir, Remdesivir). As suggested by the evaluation in Table 2, Model I had better performance and was therefore used in this study".
|
19 |
|
|
|
20 |
|
21 |
+
| **Drug** | **Model I: Original Tweet** | | | **Model II: Drug Names Masked** | | |
|
22 |
+
|------------------------|:---------------------------:|:-----------:|:------------:|:-------------------------------:|:-----------:|:------------:|
|
23 |
+
| | **Precision** | **Recall** | **F1-Score** | **Precision** | **Recall** | **F1-Score** |
|
24 |
+
| **Hydroxychloroquine** | 0.93 | 0.92 | **0.92** | 0.84 | 0.83 | 0.83 |
|
25 |
+
| **Ivermectin** | 0.92 | 0.91 | **0.91** | 0.72 | 0.68 | 0.68 |
|
26 |
+
| **Molnupiravir** | 0.89 | 0.89 | **0.89** | 0.78 | 0.77 | 0.77 |
|
27 |
+
| **Remdesivir** | 0.82 | 0.79 | **0.79** | 0.70 | 0.66 | 0.66 |
|
28 |
|
29 |
+
The model uploaded here is Model I.
|
30 |
+
|
31 |
+
## Intended uses & limitations
|
32 |
+
Predict opinions (negative, neutral, and positive) of tweet initiators regarding the use of a drug for the treatment or prevention of the coronavirus.
|
33 |
|
34 |
+
The model works better when there's only one drug name in a single tweet.
|
35 |
+
## Training and evaluation data
|
36 |
+
COVID-CQ
|
37 |
|
38 |
## Training procedure
|
39 |
+
See our Github repo for training scripts. [https://github.com/ningkko/COVID-drug](https://github.com/ningkko/COVID-drug)
|
40 |
### Training hyperparameters
|
41 |
|
42 |
The following hyperparameters were used during training:
|
43 |
- learning_rate: 5e-05
|
44 |
+
- train_batch_size: 24
|
45 |
+
- eval_batch_size: 24
|
46 |
- seed: 42
|
47 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
48 |
- lr_scheduler_type: linear
|
49 |
+
- num_epochs: 3.0
|
|
|
50 |
|
51 |
### Framework versions
|
52 |
|