Skip to content

Commit 0dbd569

Browse files
committed
add stemming option
1 parent ed6d986 commit 0dbd569

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

files2rouge/files2rouge.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ def run(summ_path,
2929
verbose=False,
3030
saveto=None,
3131
eos=".",
32-
ignore_empty=False):
32+
ignore_empty=False,
33+
stemming=False):
3334

3435
if saveto is not None:
3536
saveto = open(saveto, 'w')
@@ -66,6 +67,9 @@ def run(summ_path,
6667
'-r', 1000,
6768
'-n', 2,
6869
'-a']
70+
if stemming:
71+
rouge_args.append("-m")
72+
6973
rouge_args_str = " ".join([str(_) for _ in rouge_args])
7074
else:
7175
rouge_args_str = rouge_args
@@ -79,8 +83,8 @@ def run(summ_path,
7983
def main():
8084
parser = argparse.ArgumentParser(
8185
description="Calculating ROUGE score between two files (line-by-line)")
82-
parser.add_argument("summary", help="Path of summary file")
8386
parser.add_argument("reference", help="Path of references file")
87+
parser.add_argument("summary", help="Path of summary file")
8488
parser.add_argument('-v', '--verbose', action="store_true",
8589
help="""Prints ROUGE logs""")
8690
parser.add_argument('-a', '--args', help="ROUGE Arguments")
@@ -89,6 +93,7 @@ def main():
8993
parser.add_argument('-e', '--eos', dest="eos", default='.',
9094
help="""End of sentence separator (for multisentence).
9195
Default: \".\" """)
96+
parser.add_argument("-m", "--stemming", action="store_true")
9297
parser.add_argument("-i", "--ignore_empty", action="store_true")
9398
args = parser.parse_args()
9499

@@ -98,7 +103,8 @@ def main():
98103
args.verbose,
99104
args.saveto,
100105
args.eos,
101-
args.ignore_empty)
106+
args.ignore_empty,
107+
args.stemming)
102108

103109

104110
if __name__ == '__main__':

0 commit comments

Comments
 (0)