# pip install -U transformers accelerate
# 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", device_map="auto")Quick Links
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.
- Downloads last month
- 11
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="sschet/bert-base-uncased_clinical-ner")