Skip to content

Commit a3e5532

Browse files
committed
fixed error in model eval
1 parent bb54c4a commit a3e5532

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

notebooks/seq2seq_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -413,8 +413,8 @@ def evaluate_model(self, holdout_bodies, holdout_titles):
413413
for issue_body, issue_title in zip(holdout_bodies, holdout_titles):
414414
_, yhat = self.generate_issue_title(issue_body)
415415

416-
actual.append(self.pp_title.process_text(issue_title))
417-
predicted.append(self.pp_title.process_text(yhat))
416+
actual.append(self.pp_title.process_text([issue_title])[0])
417+
predicted.append(self.pp_title.process_text([yhat])[0])
418418
# calculate BLEU score
419419
bleu = corpus_bleu(actual, predicted)
420420
return bleu

0 commit comments

Comments
 (0)