MLPACK  1.0.10
Public Member Functions | Private Attributes | List of all members
mlpack::amf::SVDBatchLearning Class Reference

Public Member Functions

 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...
 

Private Attributes

double kh
 
double kw
 
double max
 
arma::mat mH
 
double min
 
double momentum
 
arma::mat mW
 
double u
 

Detailed Description

Definition at line 29 of file svd_batch_learning.hpp.

Constructor & Destructor Documentation

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

Definition at line 32 of file svd_batch_learning.hpp.

Member Function Documentation

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
VInput matrix to be factorized.
WBasis matrix.
HEncoding 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

Definition at line 42 of file svd_batch_learning.hpp.

References mH, and mW.

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
VInput matrix to be factorized.
WBasis matrix to be updated.
HEncoding matrix.

Definition at line 61 of file svd_batch_learning.hpp.

References kw, momentum, mW, and u.

Member Data Documentation

double mlpack::amf::SVDBatchLearning::kh
private

Definition at line 134 of file svd_batch_learning.hpp.

Referenced by HUpdate().

double mlpack::amf::SVDBatchLearning::kw
private

Definition at line 133 of file svd_batch_learning.hpp.

Referenced by WUpdate().

double mlpack::amf::SVDBatchLearning::max
private

Definition at line 136 of file svd_batch_learning.hpp.

arma::mat mlpack::amf::SVDBatchLearning::mH
private

Definition at line 140 of file svd_batch_learning.hpp.

Referenced by HUpdate(), and Initialize().

double mlpack::amf::SVDBatchLearning::min
private

Definition at line 135 of file svd_batch_learning.hpp.

double mlpack::amf::SVDBatchLearning::momentum
private

Definition at line 137 of file svd_batch_learning.hpp.

Referenced by HUpdate(), and WUpdate().

arma::mat mlpack::amf::SVDBatchLearning::mW
private

Definition at line 139 of file svd_batch_learning.hpp.

Referenced by Initialize(), and WUpdate().

double mlpack::amf::SVDBatchLearning::u
private

Definition at line 132 of file svd_batch_learning.hpp.

Referenced by HUpdate(), and WUpdate().


The documentation for this class was generated from the following file: