svd_wrapper.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_SVDWRAPPER_HPP
#define MLPACK_METHODS_SVDWRAPPER_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack
{
namespace cf
{

class DummyClass {};

template<class Factorizer = DummyClass>
class SVDWrapper
{
 public:
  // empty constructor
  SVDWrapper(const Factorizer& factorizer = Factorizer()) :
      factorizer(factorizer)
  {
    // Nothing to do here.
  }

  double Apply(const arma::mat& V,
               arma::mat& W,
               arma::mat& sigma,
               arma::mat& H) const;
  double Apply(const arma::mat& V,
               size_t r,
               arma::mat& W,
               arma::mat& H) const;

 private:
  Factorizer factorizer;
}; // class SVDWrapper

typedef SVDWrapper<DummyClass> ArmaSVDFactorizer;

#include "svd_wrapper_impl.hpp"

} // namespace cf
} // namespace mlpack

#endif