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("{}\t{}".format(id2tag[pred], txt)
|
| 56 |
-
# B_MT
|
| 57 |
-
# B_MT
|
| 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 |
```
|