nmf_method.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_CF_DECOMPOSITION_POLICIES_NMF_METHOD_HPP
#define MLPACK_METHODS_CF_DECOMPOSITION_POLICIES_NMF_METHOD_HPP

#include <mlpack/prereqs.hpp>
#include <mlpack/methods/amf/amf.hpp>
#include <mlpack/methods/amf/update_rules/nmf_als.hpp>
#include <mlpack/methods/amf/termination_policies/max_iteration_termination.hpp>
#include <mlpack/methods/amf/termination_policies/simple_residue_termination.hpp>

namespace mlpack {
namespace cf {

class NMFPolicy
{
 public:
  template<typename MatType>
  void Apply(const MatType& /* data */,
             const arma::sp_mat& cleanedData,
             const size_t rank,
             arma::mat& w,
             arma::mat& h,
             const size_t maxIterations,
             const double minResidue,
             const bool mit)
  {
    if (mit)
    {
      amf::MaxIterationTermination iter(maxIterations);

      // Do singular value decomposition using the NMF algorithm.
      amf::AMF<amf::MaxIterationTermination, amf::RandomInitialization,
          amf::NMFALSUpdate> nmf(iter);
      nmf.Apply(cleanedData, rank, w, h);
    }
    else
    {
      amf::SimpleResidueTermination srt(minResidue, maxIterations);

      // Do singular value decomposition using the NMF algorithm.
      amf::NMFALSFactorizer nmf(srt);
      nmf.Apply(cleanedData, rank, w, h);
    }
  }
};

} // namespace cf
} // namespace mlpack

#endif