kornosk commited on
Commit
b5888f7
1 Parent(s): 7ae21e4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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)