Skip to content

Commit 4a9acfb

Browse files
committed
Fixed rouge
1 parent 63dce6c commit 4a9acfb

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

model.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,8 +225,9 @@ def evaluate(self, release=False):
225225

226226
elapsed = int(time.time() - eval_start_time)
227227
precision, recall, f1 = self.calculate_results(true_positive, false_positive, false_negative)
228-
files_rouge = FilesRouge(predicted_file_name, ref_file_name)
229-
rouge = files_rouge.get_scores(avg=True, ignore_empty=True)
228+
files_rouge = FilesRouge()
229+
rouge = files_rouge.get_scores(
230+
hyp_path=predicted_file_name, ref_path=ref_file_name, avg=True, ignore_empty=True)
230231
print("Evaluation time: %sh%sm%ss" % ((elapsed // 60 // 60), (elapsed // 60) % 60, elapsed % 60))
231232
return num_correct_predictions / total_predictions, \
232233
precision, recall, f1, rouge

0 commit comments

Comments
 (0)