Downloads · 30 days
6
6% of all-time downloads
sidupadhyay/bert-base-uncased_clinical-ner-ONNX
bert-base-uncased_clinical-ner-ONNX is a token classification model from sidupadhyay. Use it when you need labels on individual words, such as names. It is set up for onnxruntime. The card lists the license as apache-2.0.
Available on Hugging Face: sidupadhyay/bert-base-uncasedclinical-ner-ONNX
Downloads · 30 days
6
6% of all-time downloads
All-time downloads
102
Public
Repo size
436 MB
Likes
0
Public
Click a slice to open those files.
.onnx436 MB · 100%
From the Hugging Face model README
Available on Hugging Face: sidupadhyay/bert-base-uncased_clinical-ner-ONNX
This is an ONNX export of samrawal/bert-base-uncased_clinical-ner.
All credit for model training and development goes to the original author, samrawal.
| Label | Description |
|---|---|
B-problem / I-problem | Clinical problem or diagnosis |
B-treatment / I-treatment | Treatment or medication |
B-test / I-test | Clinical test or procedure |
O | Outside any entity |
from transformers import AutoTokenizer
from optimum.onnxruntime import ORTModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("sidupadhyay/bert-base-uncased_clinical-ner-ONNX")
model = ORTModelForTokenClassification.from_pretrained("sidupadhyay/bert-base-uncased_clinical-ner-ONNX")
inputs = tokenizer("The patient was prescribed aspirin for chest pain.", return_tensors="pt")
outputs = model(**inputs)
For more details on training data, evaluation, and intended use, see the original model card: samrawal/bert-base-uncased_clinical-ner.