Update README.md
Browse files
README.md
CHANGED
@@ -8,11 +8,11 @@ This model is pre-trained on over 5 million English tweets about the 2020 US Pre
|
|
8 |
|
9 |
# Training Objective
|
10 |
|
11 |
-
This model is initialized with BERT-base and trained with normal MLM objective with classification layer fine-tuned for stance detection.
|
12 |
|
13 |
# Usage
|
14 |
|
15 |
-
This pre-trained language model is fine-tuned to the stance detection task.
|
16 |
|
17 |
Please see the [official repository](https://github.com/GU-DataLab/stance-detection-KE-MLM) for more detail.
|
18 |
|
@@ -25,7 +25,7 @@ import numpy as np
|
|
25 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
26 |
|
27 |
# select mode path here
|
28 |
-
pretrained_LM_path = "kornosk/bert-election2020-twitter-stance"
|
29 |
|
30 |
# load model
|
31 |
tokenizer = AutoTokenizer.from_pretrained(pretrained_LM_path)
|
|
|
8 |
|
9 |
# Training Objective
|
10 |
|
11 |
+
This model is initialized with BERT-base and trained with normal MLM objective with classification layer fine-tuned for stance detection towards Joe Biden.
|
12 |
|
13 |
# Usage
|
14 |
|
15 |
+
This pre-trained language model is fine-tuned to the stance detection task specifically for Joe Biden.
|
16 |
|
17 |
Please see the [official repository](https://github.com/GU-DataLab/stance-detection-KE-MLM) for more detail.
|
18 |
|
|
|
25 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
26 |
|
27 |
# select mode path here
|
28 |
+
pretrained_LM_path = "kornosk/bert-election2020-twitter-stance-biden"
|
29 |
|
30 |
# load model
|
31 |
tokenizer = AutoTokenizer.from_pretrained(pretrained_LM_path)
|