|
mlpack
|
Implementation of the negative log likelihood layer. More...
#include <negative_log_likelihood.hpp>
Public Member Functions | |
| NegativeLogLikelihood () | |
| Create the NegativeLogLikelihoodLayer object. | |
| template<typename PredictionType , typename TargetType > | |
| PredictionType::elem_type | Forward (const PredictionType &prediction, const TargetType &target) |
| Computes the Negative log likelihood. More... | |
| template<typename PredictionType , typename TargetType , typename LossType > | |
| void | Backward (const PredictionType &prediction, const TargetType &target, LossType &loss) |
| Ordinary feed backward pass of a neural network. More... | |
| InputDataType & | InputParameter () const |
| Get the input parameter. | |
| InputDataType & | InputParameter () |
| Modify the input parameter. | |
| OutputDataType & | OutputParameter () const |
| Get the output parameter. | |
| OutputDataType & | OutputParameter () |
| Modify the output parameter. | |
| OutputDataType & | Delta () const |
| Get the delta. | |
| OutputDataType & | Delta () |
| Modify the delta. | |
| template<typename Archive > | |
| void | serialize (Archive &, const uint32_t) |
| Serialize the layer. | |
Implementation of the negative log likelihood layer.
The negative log likelihood layer expectes that the input contains log-probabilities for each class. The layer also expects a class index, in the range between 1 and the number of classes, as target when calling the Forward function.
| 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). |
| void mlpack::ann::NegativeLogLikelihood< InputDataType, OutputDataType >::Backward | ( | const PredictionType & | prediction, |
| const TargetType & | target, | ||
| LossType & | loss | ||
| ) |
Ordinary feed backward pass of a neural network.
The negative log likelihood layer expects that the input contains log-probabilities for each class. The layer also expects a class index, in the range between 1 and the number of classes, as target when calling the Forward function.
| prediction | Predictions used for evaluating the specified loss function. |
| target | The target vector, that contains the class index in the range between 1 and the number of classes. |
| loss | The calculated error. |
| PredictionType::elem_type mlpack::ann::NegativeLogLikelihood< InputDataType, OutputDataType >::Forward | ( | const PredictionType & | prediction, |
| const TargetType & | target | ||
| ) |
Computes the Negative log likelihood.
| iprediction | Predictions used for evaluating the specified loss function. |
| target | The target vector, that contains the class index in the range between 1 and the number of classes. |
1.8.13