|
| SVDBatchLearning (double u=0.0002, double kw=0, double kh=0, double momentum=0.9, double min=-DBL_MIN, double max=DBL_MAX) |
|
template<typename MatType > |
void | HUpdate (const MatType &V, const arma::mat &W, arma::mat &H) |
| The update rule for the encoding matrix H. More...
|
|
template<typename MatType > |
void | Initialize (const MatType &dataset, const size_t rank) |
|
template<typename MatType > |
void | WUpdate (const MatType &V, arma::mat &W, const arma::mat &H) |
| The update rule for the basis matrix W. More...
|
|
Definition at line 29 of file svd_batch_learning.hpp.
mlpack::amf::SVDBatchLearning::SVDBatchLearning |
( |
double |
u = 0.0002 , |
|
|
double |
kw = 0 , |
|
|
double |
kh = 0 , |
|
|
double |
momentum = 0.9 , |
|
|
double |
min = -DBL_MIN , |
|
|
double |
max = DBL_MAX |
|
) |
| |
|
inline |
template<typename MatType >
void mlpack::amf::SVDBatchLearning::HUpdate |
( |
const MatType & |
V, |
|
|
const arma::mat & |
W, |
|
|
arma::mat & |
H |
|
) |
| |
|
inline |
The update rule for the encoding matrix H.
The function takes in all the matrices and only changes the value of the H matrix.
- Parameters
-
V | Input matrix to be factorized. |
W | Basis matrix. |
H | Encoding matrix to be updated. |
Definition at line 101 of file svd_batch_learning.hpp.
References kh, mH, momentum, and u.
template<typename MatType >
void mlpack::amf::SVDBatchLearning::Initialize |
( |
const MatType & |
dataset, |
|
|
const size_t |
rank |
|
) |
| |
|
inline |
template<typename MatType >
void mlpack::amf::SVDBatchLearning::WUpdate |
( |
const MatType & |
V, |
|
|
arma::mat & |
W, |
|
|
const arma::mat & |
H |
|
) |
| |
|
inline |
The update rule for the basis matrix W.
The function takes in all the matrices and only changes the value of the W matrix.
- Parameters
-
V | Input matrix to be factorized. |
W | Basis matrix to be updated. |
H | Encoding matrix. |
Definition at line 61 of file svd_batch_learning.hpp.
References kw, momentum, mW, and u.
double mlpack::amf::SVDBatchLearning::kh |
|
private |
double mlpack::amf::SVDBatchLearning::kw |
|
private |
double mlpack::amf::SVDBatchLearning::max |
|
private |
arma::mat mlpack::amf::SVDBatchLearning::mH |
|
private |
double mlpack::amf::SVDBatchLearning::min |
|
private |
double mlpack::amf::SVDBatchLearning::momentum |
|
private |
arma::mat mlpack::amf::SVDBatchLearning::mW |
|
private |
double mlpack::amf::SVDBatchLearning::u |
|
private |
The documentation for this class was generated from the following file: