Token Classification
Transformers
Safetensors
English
bert
named-entity-recognition
biomedical-nlp
disease-entity-recognition
medical-diagnosis
pathology
biocuration
disease
Instructions to use OpenMed/OpenMed-NER-DiseaseDetect-PubMed-335M with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use OpenMed/OpenMed-NER-DiseaseDetect-PubMed-335M with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="OpenMed/OpenMed-NER-DiseaseDetect-PubMed-335M")# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("OpenMed/OpenMed-NER-DiseaseDetect-PubMed-335M") model = AutoModelForTokenClassification.from_pretrained("OpenMed/OpenMed-NER-DiseaseDetect-PubMed-335M", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 196 Bytes
b9c65c7 | 1 2 3 4 5 6 7 | {
"eval_accuracy": 0.9848751359733376,
"eval_f1": 0.9096947935368043,
"eval_loss": 0.32842814922332764,
"eval_precision": 0.8931782125859333,
"eval_recall": 0.926833729650631
} |