Update rouge_metric.py

Yale-LILY · Apr 21, 2021 · e895502 · e895502
1 parent 2d14f8b
commit e895502
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/evaluation/summ_eval/rouge_metric.py b/evaluation/summ_eval/rouge_metric.py
@@ -17,7 17,7 @@
 
 @gin.configurable
 class RougeMetric(Metric):
- def __init__(self, rouge_dir=ROUGE_HOME, rouge_args=None, verbose=False):
  def __init__(self, rouge_dir=ROUGE_HOME, rouge_args=None):
  """
  ROUGE metric
  Makes use of pyrouge: https://github.com/bheinzerling/pyrouge
@@ -29,8 29,7 @@ def __init__(self, rouge_dir=ROUGE_HOME, rouge_args=None, verbose=False):
 
  """
  # 
- log_level = logging.ERROR if not verbose else None
- self.r = Rouge155(rouge_dir=rouge_dir, rouge_args=rouge_args, log_level=log_level)
  self.r = Rouge155(rouge_dir=rouge_dir, rouge_args=rouge_args)
  self.rouge_args = rouge_args
 
  def evaluate_example(self, summary, reference):