mlpack
master
|
The mean squared error performance function measures the network's performance according to the mean of squared errors. More...
Public Member Functions | |
MeanSquaredError () | |
Create the MeanSquaredError object. More... | |
template<typename eT > | |
void | Backward (const arma::Mat< eT > &&input, const arma::Mat< eT > &&target, arma::Mat< eT > &&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 eT > | |
double | Forward (const arma::Mat< eT > &&input, const arma::Mat< eT > &&target) |
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 &ar, const unsigned int) |
Serialize the layer. More... | |
Private Attributes | |
OutputDataType | delta |
Locally-stored delta object. More... | |
InputDataType | inputParameter |
Locally-stored input parameter object. More... | |
OutputDataType | outputParameter |
Locally-stored output parameter object. More... | |
The mean squared error performance function measures the network's performance according to the mean of squared errors.
ActivationFunction | Activation function used for the embedding layer. |
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 mean_squared_error.hpp.
mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::MeanSquaredError | ( | ) |
Create the MeanSquaredError object.
void mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::Backward | ( | const arma::Mat< eT > && | input, |
const arma::Mat< eT > && | target, | ||
arma::Mat< eT > && | output | ||
) |
Ordinary feed backward pass of a neural network.
input | The propagated input activation. |
target | The target vector. |
output | The calculated error. |
|
inline |
Get the delta.
Definition at line 73 of file mean_squared_error.hpp.
References mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::delta.
|
inline |
Modify the delta.
Definition at line 75 of file mean_squared_error.hpp.
References mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::delta, and mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::Serialize().
double mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::Forward | ( | const arma::Mat< eT > && | input, |
const arma::Mat< eT > && | target | ||
) |
|
inline |
Get the input parameter.
Definition at line 63 of file mean_squared_error.hpp.
References mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::inputParameter.
|
inline |
Modify the input parameter.
Definition at line 65 of file mean_squared_error.hpp.
References mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::inputParameter.
|
inline |
Get the output parameter.
Definition at line 68 of file mean_squared_error.hpp.
References mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::outputParameter.
|
inline |
Modify the output parameter.
Definition at line 70 of file mean_squared_error.hpp.
References mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::outputParameter.
void mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::Serialize | ( | Archive & | ar, |
const unsigned | int | ||
) |
Serialize the layer.
Referenced by mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::Delta().
|
private |
Locally-stored delta object.
Definition at line 85 of file mean_squared_error.hpp.
Referenced by mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::Delta().
|
private |
Locally-stored input parameter object.
Definition at line 88 of file mean_squared_error.hpp.
Referenced by mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::InputParameter().
|
private |
Locally-stored output parameter object.
Definition at line 91 of file mean_squared_error.hpp.
Referenced by mlpack::ann::MeanSquaredError< InputDataType, OutputDataType >::OutputParameter().