ckb commited on
Commit
e7eae00
·
1 Parent(s): f95c10c
Files changed (1) hide show
  1. unigram.py +1 -1
unigram.py CHANGED
@@ -97,7 +97,7 @@ class unigram(evaluate.Metric):
97
  def _compute(self, predictions, references):
98
  """Returns the scores"""
99
  # TODO: Compute the different scores of the module
100
- score = self._prec_recall_f1_score(i,j)
101
  return {
102
  "precision": score[0],
103
  "recall": score[1],
 
97
  def _compute(self, predictions, references):
98
  """Returns the scores"""
99
  # TODO: Compute the different scores of the module
100
+ score = self._prec_recall_f1_score(predictions, references)
101
  return {
102
  "precision": score[0],
103
  "recall": score[1],