Implementation of the negative log likelihood layer. More...
Public Member Functions | |
NegativeLogLikelihood () | |
Create the NegativeLogLikelihoodLayer 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... | |
OutputDataType & | Delta () const |
Get the delta. More... | |
OutputDataType & | Delta () |
Modify the delta. More... | |
template < typename InputType , typename TargetType > | |
double | Forward (const InputType &&input, TargetType &&target) |
Computes the Negative log likelihood. More... | |
InputDataType & | InputParameter () const |
Get the input parameter. More... | |
InputDataType & | InputParameter () |
Modify the input parameter. More... | |
OutputDataType & | OutputParameter () const |
Get the output parameter. More... | |
OutputDataType & | OutputParameter () |
Modify the output parameter. More... | |
template < typename Archive > | |
void | serialize (Archive &, const unsigned int) |
Serialize the layer. More... | |
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). |
Definition at line 35 of file negative_log_likelihood.hpp.
Create the NegativeLogLikelihoodLayer object.
void Backward | ( | const InputType && | input, |
const TargetType && | target, | ||
OutputType && | output | ||
) |
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.
input | The propagated input activation. |
target | The target vector, that contains the class index in the range between 1 and the number of classes. |
output | The calculated error. |
|
inline |
Get the delta.
Definition at line 80 of file negative_log_likelihood.hpp.
|
inline |
Modify the delta.
Definition at line 82 of file negative_log_likelihood.hpp.
References NegativeLogLikelihood< InputDataType, OutputDataType >::serialize().
double Forward | ( | const InputType && | input, |
TargetType && | target | ||
) |
Computes the Negative log likelihood.
input | Input data used for evaluating the specified function. |
target | The target vector, that contains the class index in the range between 1 and the number of classes. |
|
inline |
Get the input parameter.
Definition at line 70 of file negative_log_likelihood.hpp.
|
inline |
Modify the input parameter.
Definition at line 72 of file negative_log_likelihood.hpp.
|
inline |
Get the output parameter.
Definition at line 75 of file negative_log_likelihood.hpp.
|
inline |
Modify the output parameter.
Definition at line 77 of file negative_log_likelihood.hpp.
void serialize | ( | Archive & | , |
const unsigned | int | ||
) |
Serialize the layer.
Referenced by NegativeLogLikelihood< InputDataType, OutputDataType >::Delta().