Gagan Bhatia commited on
Commit
4c54605
·
1 Parent(s): 47e276a

Update evaluate_model.py

Browse files
Files changed (1) hide show
  1. src/models/evaluate_model.py +1 -0
src/models/evaluate_model.py CHANGED
@@ -13,3 +13,4 @@ def evaluate_model():
13
  results = model.evaluate(test_df=test_df,metrics="rouge")
14
  with dagshub.dagshub_logger() as logger:
15
  logger.log_metrics(results)
 
 
13
  results = model.evaluate(test_df=test_df,metrics="rouge")
14
  with dagshub.dagshub_logger() as logger:
15
  logger.log_metrics(results)
16
+ return results