tner/bc5cdr
Viewer • Updated • 16.4k • 1.98k • 5
How to use sschet/bert-base-uncased_clinical-ner with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="sschet/bert-base-uncased_clinical-ner") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("sschet/bert-base-uncased_clinical-ner")
model = AutoModelForTokenClassification.from_pretrained("sschet/bert-base-uncased_clinical-ner")A Named Entity Recognition model for clinical entities (problem, treatment, test)
The model has been trained on the i2b2 (now n2c2) dataset for the 2010 - Relations task. Please visit the n2c2 site to request access to the dataset.