ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
losses.mse.MeanSquaredError Class Reference
Collaboration diagram for losses.mse.MeanSquaredError:

Public Member Functions

 __init__ (self, reduction="mean")
 

Additional Inherited Members

- Data Fields inherited from losses.loss.LossFunction
 reduction
 

Detailed Description

Mean Squared Error (MSE) Loss Function with reduction type.

Definition at line 5 of file mse.py.

Constructor & Destructor Documentation

◆ __init__()

losses.mse.MeanSquaredError.__init__ (   self,
  reduction = "mean" 
)
Initialize the MSE loss function.
Args:
    reduction (str): Reduction type ('mean', 'sum').

Reimplemented from losses.loss.LossFunction.

Definition at line 9 of file mse.py.

9 def __init__(self, reduction="mean"):
10 """
11 Initialize the MSE loss function.
12 Args:
13 reduction (str): Reduction type ('mean', 'sum').
14 """
15 super().__init__(reduction)

References losses.mse.MeanSquaredError.__init__().

Referenced by losses.mse.MeanSquaredError.__init__().


The documentation for this class was generated from the following file: