Gagan Bhatia commited on
Commit
5ca9da8
1 Parent(s): 9621282

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
@@ -18,3 +18,4 @@ 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
 
 
 
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: