gagan3012 commited on
Commit
10a5b65
·
1 Parent(s): 96c4203

logging added

Browse files
Files changed (1) hide show
  1. src/models/evaluate_model.py +4 -0
src/models/evaluate_model.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from src.models.model import Summarization
2
  from src.data.make_dataset import make_dataset
3
 
@@ -9,4 +11,6 @@ def evaluate_model():
9
  model = Summarization()
10
  model.load_model()
11
  results = model.evaluate(test_df=test_df)
 
 
12
  return results
 
1
+ import dagshub
2
+
3
  from src.models.model import Summarization
4
  from src.data.make_dataset import make_dataset
5
 
 
11
  model = Summarization()
12
  model.load_model()
13
  results = model.evaluate(test_df=test_df)
14
+ with dagshub.dagshub_logger() as logger:
15
+ logger.log_metrics(results)
16
  return results