gagan3012 commited on
Commit
940c82c
·
1 Parent(s): 4f3ee24
Files changed (1) hide show
  1. src/models/evaluate_model.py +7 -3
src/models/evaluate_model.py CHANGED
@@ -3,6 +3,7 @@ import yaml
3
 
4
  from model import Summarization
5
  import pandas as pd
 
6
 
7
 
8
  def evaluate_model():
@@ -17,6 +18,9 @@ def evaluate_model():
17
  model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
18
  results = model.evaluate(test_df=test_df, metrics=params['metric'])
19
 
20
- with dagshub.dagshub_logger() as logger:
21
- logger.log_metrics(results)
22
- return results
 
 
 
 
3
 
4
  from model import Summarization
5
  import pandas as pd
6
+ import json
7
 
8
 
9
  def evaluate_model():
 
18
  model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
19
  results = model.evaluate(test_df=test_df, metrics=params['metric'])
20
 
21
+ with open('reports/metrics.txt', 'w') as fp:
22
+ json.dump(results, fp)
23
+
24
+
25
+ if __name__ == '__main__':
26
+ evaluate_model()