Elemets commited on
Commit
edee94e
·
verified ·
1 Parent(s): b135327

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +20 -1
README.md CHANGED
@@ -1,3 +1,22 @@
 
1
  ---
2
- license: apache-2.0
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
  ---
3
+ license: mit
4
+ base_model: emilyalsentzer/Bio_ClinicalBERT
5
+ pipeline_tag: text-classification
6
+ tags: [multi-label, clinical, overdose]
7
  ---
8
+
9
+ # DrugDetectionBERT
10
+
11
+ Bio_ClinicalBERT fine-tuned for multi-label drug detection in death investigation narratives.
12
+
13
+ Labels: Methamphetamine, Heroin, Cocaine, Fentanyl, Alcohol, Prescription.opioids,
14
+ Any Opioids, Benzodiazepines, Others, Any Drugs.
15
+
16
+ Sigmoid outputs. Per-label thresholds in `best_thresholds.json` — do not use 0.5.
17
+
18
+ ```python
19
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
20
+ tok = AutoTokenizer.from_pretrained("Elemets/DrugDetectionBERT")
21
+ model = AutoModelForSequenceClassification.from_pretrained("Elemets/DrugDetectionBERT")
22
+