@@ -29,7 +29,8 @@ def run(summ_path,
29
29
verbose = False ,
30
30
saveto = None ,
31
31
eos = "." ,
32
- ignore_empty = False ):
32
+ ignore_empty = False ,
33
+ stemming = False ):
33
34
34
35
if saveto is not None :
35
36
saveto = open (saveto , 'w' )
@@ -66,6 +67,9 @@ def run(summ_path,
66
67
'-r' , 1000 ,
67
68
'-n' , 2 ,
68
69
'-a' ]
70
+ if stemming :
71
+ rouge_args .append ("-m" )
72
+
69
73
rouge_args_str = " " .join ([str (_ ) for _ in rouge_args ])
70
74
else :
71
75
rouge_args_str = rouge_args
@@ -79,8 +83,8 @@ def run(summ_path,
79
83
def main ():
80
84
parser = argparse .ArgumentParser (
81
85
description = "Calculating ROUGE score between two files (line-by-line)" )
82
- parser .add_argument ("summary" , help = "Path of summary file" )
83
86
parser .add_argument ("reference" , help = "Path of references file" )
87
+ parser .add_argument ("summary" , help = "Path of summary file" )
84
88
parser .add_argument ('-v' , '--verbose' , action = "store_true" ,
85
89
help = """Prints ROUGE logs""" )
86
90
parser .add_argument ('-a' , '--args' , help = "ROUGE Arguments" )
@@ -89,6 +93,7 @@ def main():
89
93
parser .add_argument ('-e' , '--eos' , dest = "eos" , default = '.' ,
90
94
help = """End of sentence separator (for multisentence).
91
95
Default: \" .\" """ )
96
+ parser .add_argument ("-m" , "--stemming" , action = "store_true" )
92
97
parser .add_argument ("-i" , "--ignore_empty" , action = "store_true" )
93
98
args = parser .parse_args ()
94
99
@@ -98,7 +103,8 @@ def main():
98
103
args .verbose ,
99
104
args .saveto ,
100
105
args .eos ,
101
- args .ignore_empty )
106
+ args .ignore_empty ,
107
+ args .stemming )
102
108
103
109
104
110
if __name__ == '__main__' :
0 commit comments