gilbert / README.md
rvignav's picture
Add metadata
1d3eebe
|
raw
history blame
No virus
415 Bytes
metadata
language:
  - py
metrics:
  - f1

To use our fine-tuned BioBERT model to remove references to priors from radiology reports, run the following:

from transformers import AutoTokenizer, AutoModelForTokenClassification

modelname = "rajpurkarlab/biobert-finetuned-prior-rmv"
tokenizer = AutoTokenizer.from_pretrained(modelname)
model = AutoModelForTokenClassification.from_pretrained(modelname)