|
| SoftMarginLoss (const bool reduction=true) |
| Create the SoftMarginLoss object. More...
|
|
template<typename InputType , typename TargetType , typename OutputType > |
void | Backward (const InputType &input, const TargetType &target, OutputType &output) |
| Ordinary feed backward pass of a neural network. More...
|
|
template<typename InputType , typename TargetType > |
InputType::elem_type | Forward (const InputType &input, const TargetType &target) |
| Computes the Soft Margin Loss function. More...
|
|
OutputDataType & | OutputParameter () |
| Modify the output parameter. More...
|
|
OutputDataType & | OutputParameter () const |
| Get the output parameter. More...
|
|
bool & | Reduction () |
| Modify the type of reduction used. More...
|
|
bool | Reduction () const |
| Get the type of reduction used. More...
|
|
template<typename Archive > |
void | serialize (Archive &ar, const unsigned int) |
| Serialize the layer. More...
|
|
template<typename InputDataType = arma::mat, typename OutputDataType = arma::mat>
class mlpack::ann::SoftMarginLoss< InputDataType, OutputDataType >
- Template Parameters
-
InputDataType | Type of the input data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
OutputDataType | Type of the output data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
Definition at line 34 of file soft_margin_loss.hpp.