src/utils/ml_utils/metric/classification_metric.py

from src.entity.artifact_entity import ClassificationMetricArtifact
from src.exception.exception import NetworkSecurityException
from sklearn.metrics import f1_score, precision_score, recall_score
import sys

def classification_score(y_true, y_pred):
    try:
        model_f1_score = f1_score(y_true, y_pred)
        model_precision_score = precision_score(y_true, y_pred)
        model_recall_score = recall_score(y_true, y_pred)   
        classification_metric = ClassificationMetricArtifact(f1_score=model_f1_score, precision_score=model_precision_score, recall_score=model_recall_score)
        return classification_metric
    except Exception as e:
        raise NetworkSecurityException(e, sys) from e