File size: 299 Bytes
6cf7502
 
 
 
 
 
 
 
45602a9
 
 
 
77c86f8
2b622e6
1
2
3
4
5
6
7
8
9
10
11
12
13
14
To use the model, add the following from the transformers package:

(1) Tokenizer:

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

(2) Model type:

#T in [1,2,3,4]


num_classes = 4 

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