Token Classification
Transformers
Safetensors
English
bert
ner
clinical
medical
healthcare
biomedical
pubmedbert
named-entity-recognition
diagnosis
icd-10
icd-coding
snomed
sapbert
entity-linking
disease
ehr
hipaa
Eval Results (legacy)
Instructions to use genzeonplatform/healthcare-brain-diagnosis-icd-ner with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use genzeonplatform/healthcare-brain-diagnosis-icd-ner with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="genzeonplatform/healthcare-brain-diagnosis-icd-ner")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("genzeonplatform/healthcare-brain-diagnosis-icd-ner") model = AutoModelForTokenClassification.from_pretrained("genzeonplatform/healthcare-brain-diagnosis-icd-ner", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "id2label": { | |
| "0": "O", | |
| "1": "B-PRIMARY_DIAGNOSIS", | |
| "2": "I-PRIMARY_DIAGNOSIS", | |
| "3": "B-SECONDARY_DIAGNOSIS", | |
| "4": "I-SECONDARY_DIAGNOSIS", | |
| "5": "B-DIFFERENTIAL_DIAGNOSIS", | |
| "6": "I-DIFFERENTIAL_DIAGNOSIS", | |
| "7": "B-COMORBIDITY", | |
| "8": "I-COMORBIDITY", | |
| "9": "B-COMPLICATION", | |
| "10": "I-COMPLICATION", | |
| "11": "B-CHRONIC_CONDITION", | |
| "12": "I-CHRONIC_CONDITION", | |
| "13": "B-ACUTE_CONDITION", | |
| "14": "I-ACUTE_CONDITION", | |
| "15": "B-DIAGNOSIS_STATUS", | |
| "16": "I-DIAGNOSIS_STATUS", | |
| "17": "B-DIAGNOSIS_DATE", | |
| "18": "I-DIAGNOSIS_DATE" | |
| }, | |
| "label2id": { | |
| "O": 0, | |
| "B-PRIMARY_DIAGNOSIS": 1, | |
| "I-PRIMARY_DIAGNOSIS": 2, | |
| "B-SECONDARY_DIAGNOSIS": 3, | |
| "I-SECONDARY_DIAGNOSIS": 4, | |
| "B-DIFFERENTIAL_DIAGNOSIS": 5, | |
| "I-DIFFERENTIAL_DIAGNOSIS": 6, | |
| "B-COMORBIDITY": 7, | |
| "I-COMORBIDITY": 8, | |
| "B-COMPLICATION": 9, | |
| "I-COMPLICATION": 10, | |
| "B-CHRONIC_CONDITION": 11, | |
| "I-CHRONIC_CONDITION": 12, | |
| "B-ACUTE_CONDITION": 13, | |
| "I-ACUTE_CONDITION": 14, | |
| "B-DIAGNOSIS_STATUS": 15, | |
| "I-DIAGNOSIS_STATUS": 16, | |
| "B-DIAGNOSIS_DATE": 17, | |
| "I-DIAGNOSIS_DATE": 18 | |
| }, | |
| "label_list": [ | |
| "O", | |
| "B-PRIMARY_DIAGNOSIS", | |
| "I-PRIMARY_DIAGNOSIS", | |
| "B-SECONDARY_DIAGNOSIS", | |
| "I-SECONDARY_DIAGNOSIS", | |
| "B-DIFFERENTIAL_DIAGNOSIS", | |
| "I-DIFFERENTIAL_DIAGNOSIS", | |
| "B-COMORBIDITY", | |
| "I-COMORBIDITY", | |
| "B-COMPLICATION", | |
| "I-COMPLICATION", | |
| "B-CHRONIC_CONDITION", | |
| "I-CHRONIC_CONDITION", | |
| "B-ACUTE_CONDITION", | |
| "I-ACUTE_CONDITION", | |
| "B-DIAGNOSIS_STATUS", | |
| "I-DIAGNOSIS_STATUS", | |
| "B-DIAGNOSIS_DATE", | |
| "I-DIAGNOSIS_DATE" | |
| ] | |
| } |