tner/bc5cdr
Viewer • Updated • 16.4k • 1.93k • 5
How to use sschet/bert-large-uncased_med-ner with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="sschet/bert-large-uncased_med-ner") # Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("sschet/bert-large-uncased_med-ner")
model = AutoModelForTokenClassification.from_pretrained("sschet/bert-large-uncased_med-ner")A Named Entity Recognition model for medication entities (medication name, dosage, duration, frequency, reason).
The model has been trained on the i2b2 (now n2c2) dataset for the 2009 - Medication task. Please visit the n2c2 site to request access to the dataset.