gagan3012 commited on
Commit
2d35df5
1 Parent(s): dec1e3d

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
@@ -13,6 +13,7 @@ def evaluate_model():
13
  params = yaml.safe_load(f)
14
 
15
  test_df = pd.read_csv("data/processed/test.csv")
 
16
  model = Summarization()
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"])
 
13
  params = yaml.safe_load(f)
14
 
15
  test_df = pd.read_csv("data/processed/test.csv")
16
+ test_df = test_df.sample(n=25, random_state=42)
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"])