MDDDDR
/

dmis_lab_biobert_v1.1_NER

Token Classification

Model card Files Files and versions Community

MDDDDR commited on Apr 28, 2024

Commit

8d2fb61

·

verified ·

1 Parent(s): dc4ee9f

Update README.md

Files changed (1) hide show

README.md +4 -4

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ tags:
 ---
 # Model Card for Model ID
-base_model : dmis-lab/biobert-v1.1
 hidden_size : 768
@@ -51,7 +51,7 @@ pred = [p for p in np.argmax(logits, axis=2)][0][1:-1]
 # Check pred
 for txt, pred in zip(tokenizer.tokenize(text), pred):
-    print(txt, id2tag[pred])
-    # mental B_MT
-    # disorder B_MT
 ```

 ---
 # Model Card for Model ID
+base_model : [dmis-lab/biobert-v1.1](https://huggingface.co/dmis-lab/biobert-v1.1)
 hidden_size : 768
 # Check pred
 for txt, pred in zip(tokenizer.tokenize(text), pred):
+    print("{}\t{}".format(id2tag[pred], txt))
+    # B_MT mental
+    # B_MT disorder
 ```