|
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().
1.8.11