Gagan Bhatia commited on
Commit
717b039
1 Parent(s): 958d7ab

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
@@ -16,6 +16,7 @@ def evaluate_model():
16
  test_df = pd.read_csv("data/processed/test.csv")[:25]
17
  model = Summarization()
18
  model.load_model(model_type=params["model_type"], model_dir=params["model_dir"])
 
19
 
20
  with open('reports/metrics.txt', 'w') as fp:
21
  json.dump(results, fp)
 
16
  test_df = pd.read_csv("data/processed/test.csv")[:25]
17
  model = Summarization()
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)