@@ -30,7 +30,7 @@ def run(summ_path,
30
30
saveto = None ,
31
31
eos = "." ,
32
32
ignore_empty = False ,
33
- stemming = False ):
33
+ stemming = True ):
34
34
s = settings .Settings ()
35
35
s ._load ()
36
36
stime = time ()
@@ -68,7 +68,7 @@ def run(summ_path,
68
68
else :
69
69
rouge_args_str = rouge_args
70
70
rouge_args_str = "%s %s" % (data_arg , rouge_args_str )
71
-
71
+
72
72
output = r .convert_and_evaluate (rouge_args = rouge_args_str )
73
73
74
74
if saveto is not None :
@@ -91,18 +91,26 @@ def main():
91
91
parser .add_argument ('-e' , '--eos' , dest = "eos" , default = '.' ,
92
92
help = """End of sentence separator (for multisentence).
93
93
Default: \" .\" """ )
94
- parser .add_argument ("-m" , "--stemming" , action = "store_true" )
94
+ parser .add_argument ("-m" , "--stemming" , action = "store_true" ,
95
+ help = "DEPRECATED: stemming is now default behavior" )
96
+ parser .add_argument ("-nm" , "--no_stemming" , action = "store_true" ,
97
+ help = "Switch off stemming" )
95
98
parser .add_argument ("-i" , "--ignore_empty" , action = "store_true" )
96
99
args = parser .parse_args ()
97
100
101
+ if args .stemming :
102
+ raise ValueError (
103
+ """files2rouge uses stemming by default so --stemming is
104
+ deprecated. You can turn it off with -nm/--no_stemming""" )
105
+
98
106
run (args .summary ,
99
107
args .reference ,
100
- args .args ,
101
- args .verbose ,
102
- args .saveto ,
103
- args .eos ,
104
- args .ignore_empty ,
105
- args .stemming )
108
+ rouge_args = args .args ,
109
+ verbose = args .verbose ,
110
+ saveto = args .saveto ,
111
+ eos = args .eos ,
112
+ ignore_empty = args .ignore_empty ,
113
+ stemming = not args .no_stemming )
106
114
107
115
108
116
if __name__ == '__main__' :
0 commit comments