Gagan Bhatia commited on
Commit
47e276a
1 Parent(s): c398d56

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
@@ -12,3 +12,4 @@ def evaluate_model():
12
  model.load_model()
13
  results = model.evaluate(test_df=test_df,metrics="rouge")
14
  with dagshub.dagshub_logger() as logger:
 
 
12
  model.load_model()
13
  results = model.evaluate(test_df=test_df,metrics="rouge")
14
  with dagshub.dagshub_logger() as logger:
15
+ logger.log_metrics(results)