Gagan Bhatia commited on
Commit
dda2ec1
1 Parent(s): f3435ae

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -455,3 +455,4 @@ class Summarization:
455
  def evaluate(
456
  self,
457
  test_df: pd.DataFrame,
 
 
455
  def evaluate(
456
  self,
457
  test_df: pd.DataFrame,
458
+ metrics: str = "rouge"