T5-Summarization / src /models /evaluate_model.py
Gagan Bhatia
Update evaluate_model.py
4c54605
raw
history blame
430 Bytes
import dagshub
from src.models.model import Summarization
import pandas as pd
def evaluate_model():
"""
Evaluate model using rouge measure
"""
test_df = pd.load_csv('../../data/processed/test.csv')
model = Summarization()
model.load_model()
results = model.evaluate(test_df=test_df,metrics="rouge")
with dagshub.dagshub_logger() as logger:
logger.log_metrics(results)
return results