Emanuela Boros
commited on
Commit
·
ddc72bb
1
Parent(s):
0990522
add confidence
Browse files- generic_nel.py +1 -1
generic_nel.py
CHANGED
@@ -120,7 +120,7 @@ class NelPipeline(Pipeline):
|
|
120 |
"""
|
121 |
wikipedia_predictions, percentages = outputs
|
122 |
results = []
|
123 |
-
for idx, wikipedia_name in enumerate(
|
124 |
# Get QID
|
125 |
qid = get_wikipedia_page_props(wikipedia_name)
|
126 |
# print(f"{wikipedia_name} -- QID: {qid}")
|
|
|
120 |
"""
|
121 |
wikipedia_predictions, percentages = outputs
|
122 |
results = []
|
123 |
+
for idx, wikipedia_name in enumerate(wikipedia_predictions):
|
124 |
# Get QID
|
125 |
qid = get_wikipedia_page_props(wikipedia_name)
|
126 |
# print(f"{wikipedia_name} -- QID: {qid}")
|