mse.py

Go to the documentation of this file.

from typing import Literal
import numpy as np
from .loss import LossFunction
 
 
class MeanSquaredError(LossFunction):
    """
    Mean Squared Error (MSE) Loss Function with reduction type.
    """
    def __init__(self, reduction: Literal["mean", "sum"] = "mean") -> None:
        """
        Initialize the MSE loss function.
        Args:
            reduction (str): Reduction type ('mean', 'sum').
        """
        super().__init__(reduction)