utils.py

"""

Utility functions for DistilBERT sentiment analysis

File: utils.py

"""

import numpy as np
from sklearn.metrics import precision_recall_fscore_support, accuracy_score

def compute_metrics(eval_pred):
    """

    Compute evaluation metrics for binary classification

    

    Args:

        eval_pred: Tuple of (predictions, labels)

        

    Returns:

        Dict with accuracy, f1, precision, recall

    """
    predictions, labels = eval_pred
    predictions = np.argmax(predictions, axis=1)
    
    precision, recall, f1, _ = precision_recall_fscore_support(
        labels, predictions, average='binary'
    )
    accuracy = accuracy_score(labels, predictions)
    
    return {
        'accuracy': accuracy,
        'f1': f1,
        'precision': precision,
        'recall': recall
    }