PrimaryGleasonBERT / README.md
jkefeli's picture
Update README.md
32255d4
|
Raw
History Blame Contribute Delete
285 Bytes

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)