jsylee
/

scibert_scivocab_uncased-finetuned-ner

Token Classification

Named Entity Recognition

Inference Endpoints

Model card Files Files and versions Community

jsylee commited on Nov 22, 2021

Commit

2df151a

•

1 Parent(s): ca88b1a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ from transformers import (AutoModelForTokenClassification,
 model_checkpoint = "jsylee/scibert_scivocab_uncased-finetuned-ner"
 model = AutoModelForTokenClassification.from_pretrained(model_checkpoint, num_labels=5,
-                                                        id2label={0: 'O', 1: 'B-DRUG', 2: 'I-DRUG', 3: 'B-AE', 4: 'I-AE'}
                                                         )
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)

 model_checkpoint = "jsylee/scibert_scivocab_uncased-finetuned-ner"
 model = AutoModelForTokenClassification.from_pretrained(model_checkpoint, num_labels=5,
+                                                        id2label={0: 'O', 1: 'B-DRUG', 2: 'I-DRUG', 3: 'B-EFFECT', 4: 'I-EFFECT'}
                                                         )
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)