MDDDDR commited on
Commit
147772a
·
verified ·
1 Parent(s): 2a4b2fa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -51,7 +51,7 @@ pred = [p for p in np.argmax(logits, axis=2)][0][1:-1]
51
 
52
  # Check pred
53
  for txt, pred in zip(tokenizer.tokenize(text), pred):
54
- print(txt, id2tag[pred])
55
- # mental B_MT
56
- # disorder B_MT
57
  ```
 
51
 
52
  # Check pred
53
  for txt, pred in zip(tokenizer.tokenize(text), pred):
54
+ print("{}\t{}".format(id2tag[pred], txt))
55
+ # B_MT mental
56
+ # B_MT disorder
57
  ```