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