filbert / README.md
rvignav's picture
Update README.md
62c1215
metadata
language:
  - py
metrics:
  - f1

To use our fine-tuned BioBERT model to predict whether a sentence from a radiology reports makes reference to priors, run the following:

from transformers import AutoTokenizer, AutoModelForTokenClassification

modelname = "rajpurkarlab/filbert"
tokenizer = AutoTokenizer.from_pretrained(modelname)
model = AutoModelForTokenClassification.from_pretrained(modelname)