given_init.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_AMF_INIT_RULES_GIVEN_INIT_HPP
#define MLPACK_METHODS_AMF_INIT_RULES_GIVEN_INIT_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace amf {

class GivenInitialization
{
 public:
  // Empty constructor required for the InitializeRule template.
  GivenInitialization() { }

  // Initialize the GivenInitialization object with the given matrices.
  GivenInitialization(const arma::mat& w, const arma::mat& h) : w(w), h(h) { }

  // Initialize the GivenInitialization object, taking control of the given
  // matrices.
  GivenInitialization(const arma::mat&& w, const arma::mat&& h) :
    w(std::move(w)),
    h(std::move(h))
  { }

  template<typename MatType>
  inline void Initialize(const MatType& /* V */,
                         const size_t /* r */,
                         arma::mat& W,
                         arma::mat& H)
  {
    // Initialize to the given matrices.
    W = w;
    H = h;
  }

  template<typename Archive>
  void Serialize(Archive& ar, const unsigned int /* version */)
  {
    ar & data::CreateNVP(w, "w");
    ar & data::CreateNVP(h, "h");
  }

 private:
  arma::mat w;
  arma::mat h;
};

} // namespace amf
} // namespace mlpack

#endif