mlpack
master
|
This class is used to initialize the weight matrix with the Nguyen-Widrow method. More...
Public Member Functions | |
NguyenWidrowInitialization (const double lowerBound=-0.5, const double upperBound=0.5) | |
Initialize the random initialization rule with the given lower bound and upper bound. More... | |
template<typename eT > | |
void | Initialize (arma::Mat< eT > &W, const size_t rows, const size_t cols) |
Initialize the elements of the specified weight matrix with the Nguyen-Widrow method. More... | |
template<typename eT > | |
void | Initialize (arma::Cube< eT > &W, const size_t rows, const size_t cols, const size_t slices) |
Initialize the elements of the specified weight 3rd order tensor with the Nguyen-Widrow method. More... | |
Private Attributes | |
const double | lowerBound |
The number used as lower bound. More... | |
const double | upperBound |
The number used as upper bound. More... | |
This class is used to initialize the weight matrix with the Nguyen-Widrow method.
The method is defined by
Where H is the number of neurons in the outgoing layer, I represents the number of neurons in the ingoing layer and gamma defines the random interval that is used to initialize the weights with a random value in a specific range.
Definition at line 52 of file nguyen_widrow_init.hpp.
|
inline |
Initialize the random initialization rule with the given lower bound and upper bound.
lowerBound | The number used as lower bound. |
upperBound | The number used as upper bound. |
Definition at line 62 of file nguyen_widrow_init.hpp.
|
inline |
Initialize the elements of the specified weight matrix with the Nguyen-Widrow method.
W | Weight matrix to initialize. |
rows | Number of rows. |
cols | Number of columns. |
Definition at line 75 of file nguyen_widrow_init.hpp.
References mlpack::ann::RandomInitialization::Initialize(), lowerBound, and upperBound.
Referenced by Initialize().
|
inline |
Initialize the elements of the specified weight 3rd order tensor with the Nguyen-Widrow method.
W | Weight matrix to initialize. |
rows | Number of rows. |
cols | Number of columns. |
slices | Number of slices. |
Definition at line 94 of file nguyen_widrow_init.hpp.
References Initialize().
|
private |
The number used as lower bound.
Definition at line 107 of file nguyen_widrow_init.hpp.
Referenced by Initialize().
|
private |
The number used as upper bound.
Definition at line 110 of file nguyen_widrow_init.hpp.
Referenced by Initialize().