![]() |
NeuZephyr
Simple DL Framework
|
Contains loss function nodes for computing various loss metrics in a machine learning model. More...
Classes | |
class | BinaryCrossEntropyNode |
Represents the Binary Cross-Entropy (BCE) loss function node in a computational graph. More... | |
class | MeanSquaredErrorNode |
Represents the Mean Squared Error (MSE) loss function node in a computational graph. More... | |
Contains loss function nodes for computing various loss metrics in a machine learning model.
The Loss
namespace provides a collection of nodes that represent different loss functions used during the training of machine learning models. These loss functions are used to evaluate the model's performance by calculating the difference between the predicted output and the true values.
This namespace includes:
MeanSquaredErrorNode
that compute loss for regression tasks.BinaryCrossEntropyNode
for computing loss in binary classification problems.Loss function nodes perform key roles in the optimization process by guiding the model to minimize errors during training. These nodes integrate with the model’s computational graph, where they compute the forward pass loss and its gradients during the backward pass.