The earth mover distance function measures the network's performance according to the Kantorovich-Rubinstein duality approximation. More...
Public Member Functions | |
EarthMoverDistance () | |
Create the EarthMoverDistance object. More... | |
template | |
void | Backward (const InputType &input, const TargetType &target, OutputType &output) |
Ordinary feed backward pass of a neural network. More... | |
template | |
InputType::elem_type | Forward (const InputType &input, const TargetType &target) |
Ordinary feed forward pass of a neural network. More... | |
OutputDataType & | OutputParameter () const |
Get the output parameter. More... | |
OutputDataType & | OutputParameter () |
Modify the output parameter. More... | |
template | |
void | serialize (Archive &ar, const unsigned int) |
Serialize the layer. More... | |
The earth mover distance function measures the network's performance according to the Kantorovich-Rubinstein duality approximation.
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 33 of file earth_mover_distance.hpp.
Create the EarthMoverDistance object.
void Backward | ( | const InputType & | input, |
const TargetType & | target, | ||
OutputType & | output | ||
) |
Ordinary feed backward pass of a neural network.
input | The propagated input activation. |
target | The target vector. |
output | The calculated error. |
InputType::elem_type Forward | ( | const InputType & | input, |
const TargetType & | target | ||
) |
Ordinary feed forward pass of a neural network.
input | Input data used for evaluating the specified function. |
target | The target vector. |
|
inline |
Get the output parameter.
Definition at line 64 of file earth_mover_distance.hpp.
|
inline |
Modify the output parameter.
Definition at line 66 of file earth_mover_distance.hpp.
References EarthMoverDistance< InputDataType, OutputDataType >::serialize().
void serialize | ( | Archive & | ar, |
const unsigned | int | ||
) |
Serialize the layer.
Referenced by EarthMoverDistance< InputDataType, OutputDataType >::OutputParameter().