MiriUll commited on
Commit
5738764
1 Parent(s): 13785d5

Update negbleurt.py

Browse files
Files changed (1) hide show
  1. negbleurt.py +1 -1
negbleurt.py CHANGED
@@ -68,6 +68,6 @@ class NegBLEURT(evaluate.Metric):
68
 
69
  scores_negbleurt = []
70
  for i in range(0, len(references), batch_size):
71
- tokenized = self.tokenizer(references[i:i+batch_size], candidates[i:i+batch_size], return_tensors='pt', padding=True, max_length=512, truncation=True)
72
  scores_negbleurt += self.model(**tokenized).logits.flatten().tolist()
73
  return {'negBLEURT': scores_negbleurt}
 
68
 
69
  scores_negbleurt = []
70
  for i in range(0, len(references), batch_size):
71
+ tokenized = self.tokenizer(references[i:i+batch_size], predictions[i:i+batch_size], return_tensors='pt', padding=True, max_length=512, truncation=True)
72
  scores_negbleurt += self.model(**tokenized).logits.flatten().tolist()
73
  return {'negBLEURT': scores_negbleurt}