cf_model.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_CF_CF_MODEL_HPP
#define MLPACK_METHODS_CF_CF_MODEL_HPP

#include <mlpack/core.hpp>
#include <boost/variant.hpp>
#include "cf.hpp"

#include <mlpack/methods/cf/decomposition_policies/batch_svd_method.hpp>
#include <mlpack/methods/cf/decomposition_policies/randomized_svd_method.hpp>
#include <mlpack/methods/cf/decomposition_policies/regularized_svd_method.hpp>
#include <mlpack/methods/cf/decomposition_policies/svd_complete_method.hpp>
#include <mlpack/methods/cf/decomposition_policies/svd_incomplete_method.hpp>
#include <mlpack/methods/cf/decomposition_policies/bias_svd_method.hpp>
#include <mlpack/methods/cf/decomposition_policies/svdplusplus_method.hpp>

namespace mlpack {
namespace cf {

class DeleteVisitor : public boost::static_visitor<void>
{
 public:
  template<typename DecompositionPolicy>
  void operator()(CFType<DecompositionPolicy>* c) const;
};

class GetValueVisitor : public boost::static_visitor<void*>
{
 public:
  template<typename DecompositionPolicy>
  void* operator()(CFType<DecompositionPolicy>* c) const;
};

template <typename NeighborSearchPolicy,
          typename InterpolationPolicy>
class PredictVisitor : public boost::static_visitor<void>
{
 private:
  const arma::Mat<size_t>& combinations;
  arma::vec& predictions;

 public:
  template<typename DecompositionPolicy>
  void operator()(CFType<DecompositionPolicy>* c) const;

  PredictVisitor(const arma::Mat<size_t>& combinations,
                 arma::vec& predictions);
};

template <typename NeighborSearchPolicy,
          typename InterpolationPolicy>
class RecommendationVisitor : public boost::static_visitor<void>
{
 private:
  const size_t numRecs;
  arma::Mat<size_t>& recommendations;
  const arma::Col<size_t>& users;
  const bool usersGiven;

 public:
  RecommendationVisitor(const size_t numRecs,
                        arma::Mat<size_t>& recommendations,
                        const arma::Col<size_t>& users,
                        const bool usersGiven);

  template<typename DecompositionPolicy>
  void operator()(CFType<DecompositionPolicy>* c) const;
};

class CFModel
{
 private:
  boost::variant<CFType<NMFPolicy>*,
                 CFType<BatchSVDPolicy>*,
                 CFType<RandomizedSVDPolicy>*,
                 CFType<RegSVDPolicy>*,
                 CFType<SVDCompletePolicy>*,
                 CFType<SVDIncompletePolicy>*,
                 CFType<BiasSVDPolicy>*,
                 CFType<SVDPlusPlusPolicy>*> cf;

 public:
  CFModel() { }

  ~CFModel();

  template<typename DecompositionPolicy>
  const CFType<DecompositionPolicy>* CFPtr() const;

  template<typename DecompositionPolicy,
           typename MatType>
  void Train(const MatType& data,
             const size_t numUsersForSimilarity,
             const size_t rank,
             const size_t maxIterations,
             const double minResidue,
             const bool mit);

  template <typename NeighborSearchPolicy,
            typename InterpolationPolicy>
  void Predict(const arma::Mat<size_t>& combinations,
               arma::vec& predictions);

  template<typename NeighborSearchPolicy,
           typename InterpolationPolicy>
  void GetRecommendations(const size_t numRecs,
                          arma::Mat<size_t>& recommendations,
                          const arma::Col<size_t>& users);

  template<typename NeighborSearchPolicy,
           typename InterpolationPolicy>
  void GetRecommendations(const size_t numRecs,
                          arma::Mat<size_t>& recommendations);

  template<typename Archive>
  void serialize(Archive& ar, const unsigned int /* version */);
};

} // namespace cf
} // namespace mlpack

// Include implementation.
#include "cf_model_impl.hpp"

#endif