Deep Learning Algorithm Implementations 1.0.0
C++ implementations of fundamental deep learning algorithms
|
Hinge Loss with autograd support. More...
#include <losses.hpp>
Public Member Functions | |
HingeLoss (const std::string &reduction="mean") | |
Constructor. | |
Variable< T > | forward (const Variable< T > &predictions, const Variable< T > &targets) override |
Forward pass: compute hinge loss. | |
![]() | |
virtual | ~AutogradLoss ()=default |
Variable< T > | operator() (const Variable< T > &predictions, const Variable< T > &targets) |
Convenience operator for computing loss. | |
Hinge Loss with autograd support.
Hinge(y_pred, y_true) = max(0, 1 - y_true * y_pred)
Commonly used for SVM and margin-based classification.
Definition at line 176 of file losses.hpp.
|
inlineexplicit |
Constructor.
reduction | Type of reduction ('mean', 'sum', 'none') |
Definition at line 182 of file losses.hpp.
|
overridevirtual |
Forward pass: compute hinge loss.
predictions | Predicted values |
targets | Target values (-1 or +1) |
Implements dl::loss::AutogradLoss< T >.
Definition at line 137 of file losses.cpp.