Emanuela Boros commited on
Commit
ddc72bb
·
1 Parent(s): 0990522

add confidence

Browse files
Files changed (1) hide show
  1. 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(outputs):
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}")