Rodrigo Ferreira Rodrigues commited on
Commit
e0f7547
·
1 Parent(s): 866cd66

Correcting bug

Browse files
Files changed (1) hide show
  1. keywords_evaluate.py +1 -1
keywords_evaluate.py CHANGED
@@ -104,7 +104,7 @@ class Keywords_evaluate(evaluate.Metric):
104
  f_gold = set(gold)
105
  if strict:
106
  f_ans = set(find)
107
- if len(f_ans.difference(f_gold)) == 0:
108
  correct += 1
109
  else:
110
  f_ans = find[0]
 
104
  f_gold = set(gold)
105
  if strict:
106
  f_ans = set(find)
107
+ if f_ans == f_gold:
108
  correct += 1
109
  else:
110
  f_ans = find[0]