Update README.md
Browse files
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(
|
55 |
-
# mental
|
56 |
-
# disorder
|
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 |
```
|