File size: 299 Bytes
1e77f09
 
 
 
 
 
 
 
b7d24fc
 
 
 
663ea13
890ad58
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:


#N in [0,1,2,3]

num_classes = 4 

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