Gagan Bhatia commited on
Commit
fc0c429
1 Parent(s): 48357ea

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,6 +18,7 @@ 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
  json.dump(results, fp)
22
 
23
 
 
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.csv", "w") as fp:
22
  json.dump(results, fp)
23
 
24