Gagan Bhatia commited on
Commit
3e6c151
1 Parent(s): 214f2f3

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
@@ -10,3 +10,4 @@ def evaluate_model():
10
  test_df = pd.load_csv('../../data/processed/test.csv')
11
  model = Summarization()
12
  model.load_model()
 
 
10
  test_df = pd.load_csv('../../data/processed/test.csv')
11
  model = Summarization()
12
  model.load_model()
13
+ results = model.evaluate(test_df=test_df,metrics="rouge")