DrugDetectionBERT

Bio_ClinicalBERT fine-tuned for multi-label drug detection in death investigation narratives.

Labels: Methamphetamine, Heroin, Cocaine, Fentanyl, Alcohol, Prescription.opioids, Any Opioids, Benzodiazepines, Others, Any Drugs.

Sigmoid outputs. Per-label thresholds in best_thresholds.json — do not use 0.5.

from transformers import AutoTokenizer, AutoModelForSequenceClassification
tok = AutoTokenizer.from_pretrained("Elemets/DrugDetectionBERT")
model = AutoModelForSequenceClassification.from_pretrained("Elemets/DrugDetectionBERT")