summarization / src /models /evaluate_model.py
gagan3012's picture
Pipeline updates
410b92f
raw
history blame
414 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)
with dagshub.dagshub_logger() as logger:
logger.log_metrics(results)
return results