File size: 293 Bytes
4a61605
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
To use the model, add the following from the transformers package:

(1) Tokenizer:

tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-BigBird")

(2) Model type:

#M in [0,1]

num_classes = 2

model = BigBirdForSequenceClassification.from_pretrained(directory, num_labels=num_classes)