Commit
·
f9cb0bf
1
Parent(s):
3396624
confidence doesnt wokr propeerly
Browse files- generic_nel.py +2 -0
generic_nel.py
CHANGED
|
@@ -129,6 +129,8 @@ class NelPipeline(Pipeline):
|
|
| 129 |
sequence_confidence = torch.exp(log_prob_sum)
|
| 130 |
percentages = sequence_confidence.cpu().numpy() * 100.0
|
| 131 |
|
|
|
|
|
|
|
| 132 |
# Return the predictions along with the extracted entity, lOffset, and rOffset
|
| 133 |
return wikipedia_predictions, enclosed_entity, lOffset, rOffset, [percentages]
|
| 134 |
|
|
|
|
| 129 |
sequence_confidence = torch.exp(log_prob_sum)
|
| 130 |
percentages = sequence_confidence.cpu().numpy() * 100.0
|
| 131 |
|
| 132 |
+
print(wikipedia_predictions, enclosed_entity, lOffset, rOffset, [percentages])
|
| 133 |
+
|
| 134 |
# Return the predictions along with the extracted entity, lOffset, and rOffset
|
| 135 |
return wikipedia_predictions, enclosed_entity, lOffset, rOffset, [percentages]
|
| 136 |
|