MLPACK  1.0.10
Public Member Functions | Static Public Member Functions | List of all members
mlpack::amf::NMFALSUpdate Class Reference

The alternating least square update rules of matrices W and H. More...

Public Member Functions

 NMFALSUpdate ()
 
template<typename MatType >
void Initialize (const MatType &dataset, const size_t rank)
 

Static Public Member Functions

template<typename MatType >
static void HUpdate (const MatType &V, const arma::mat &W, arma::mat &H)
 The update rule for the encoding matrix H. More...
 
template<typename MatType >
static void WUpdate (const MatType &V, arma::mat &W, const arma::mat &H)
 The update rule for the basis matrix W. More...
 

Detailed Description

The alternating least square update rules of matrices W and H.

Definition at line 39 of file nmf_als.hpp.

Constructor & Destructor Documentation

mlpack::amf::NMFALSUpdate::NMFALSUpdate ( )
inline

Definition at line 43 of file nmf_als.hpp.

Member Function Documentation

template<typename MatType >
static void mlpack::amf::NMFALSUpdate::HUpdate ( const MatType &  V,
const arma::mat &  W,
arma::mat &  H 
)
inlinestatic

The update rule for the encoding matrix H.

The formula used is

\[ H = \frac{W^TV}{W^TW} \]

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 96 of file nmf_als.hpp.

template<typename MatType >
void mlpack::amf::NMFALSUpdate::Initialize ( const MatType &  dataset,
const size_t  rank 
)
inline

Definition at line 46 of file nmf_als.hpp.

template<typename MatType >
static void mlpack::amf::NMFALSUpdate::WUpdate ( const MatType &  V,
arma::mat &  W,
const arma::mat &  H 
)
inlinestatic

The update rule for the basis matrix W.

The formula used is

\[ W^T = \frac{HV^T}{HH^T} \]

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 65 of file nmf_als.hpp.


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