Emanuela Boros
commited on
Commit
•
415d945
1
Parent(s):
b1746af
added confidence
Browse files- generic_nel.py +2 -0
generic_nel.py
CHANGED
|
@@ -120,6 +120,8 @@ class NelPipeline(Pipeline):
|
|
| 120 |
wikipedia_predictions = self.tokenizer.batch_decode(
|
| 121 |
outputs, skip_special_tokens=True
|
| 122 |
)
|
|
|
|
|
|
|
| 123 |
print(wikipedia_predictions)
|
| 124 |
print(len(scores))
|
| 125 |
for score in scores:
|
|
|
|
| 120 |
wikipedia_predictions = self.tokenizer.batch_decode(
|
| 121 |
outputs, skip_special_tokens=True
|
| 122 |
)
|
| 123 |
+
print(len(outputs.sequences))
|
| 124 |
+
print(outputs.sequences.shape)
|
| 125 |
print(wikipedia_predictions)
|
| 126 |
print(len(scores))
|
| 127 |
for score in scores:
|