File size: 285 Bytes
26f7f25
 
 
 
32255d4
26f7f25
32255d4
26f7f25
32255d4
48771eb
1
2
3
4
5
6
7
8
9
10
11
To use the model, add the following from the transformers package: 

(1) ClinicalBERT tokenizer: 


tokenizer = AutoTokenizer.from_pretrained("emilyalsentzer/Bio_ClinicalBERT")

(2) Model type:

model = BertForSequenceClassification.from_pretrained(checkpoint_directory, num_labels=3)