mlpack
Classes | Typedefs | Enumerations | Functions
mlpack::cf Namespace Reference

Collaborative filtering. More...

Classes

class  AverageInterpolation
 This class performs average interpolation to generate interpolation weights for neighborhood-based collaborative filtering. More...
 
class  BatchSVDPolicy
 Implementation of the Batch SVD policy to act as a wrapper when accessing Batch SVD from within CFType. More...
 
class  BiasSVDPolicy
 Implementation of the Bias SVD policy to act as a wrapper when accessing Bias SVD from within CFType. More...
 
class  CFModel
 The model to save to disk. More...
 
class  CFType
 This class implements Collaborative Filtering (CF). More...
 
class  CFWrapper
 The CFWrapper class wraps the functionality of all CF types. More...
 
class  CFWrapperBase
 The CFWrapperBase class provides a unified interface that can be used by the CFModel class to interact with all different CF types at runtime. More...
 
class  CombinedNormalization
 This normalization class performs a sequence of normalization methods on raw ratings. More...
 
class  CosineSearch
 Nearest neighbor search with cosine distance. More...
 
class  DummyClass
 This class acts as a dummy class for passing as template parameter. More...
 
class  ItemMeanNormalization
 This normalization class performs item mean normalization on raw ratings. More...
 
class  LMetricSearch
 Nearest neighbor search with L_p distance. More...
 
class  NMFPolicy
 Implementation of the NMF policy to act as a wrapper when accessing NMF from within CFType. More...
 
class  NoNormalization
 This normalization class doesn't perform any normalization. More...
 
class  OverallMeanNormalization
 This normalization class performs overall mean normalization on raw ratings. More...
 
class  PearsonSearch
 Nearest neighbor search with pearson distance (or furthest neighbor search with pearson correlation). More...
 
class  RandomizedSVDPolicy
 Implementation of the Randomized SVD policy to act as a wrapper when accessing Randomized SVD from within CFType. More...
 
class  RegressionInterpolation
 Implementation of regression-based interpolation method. More...
 
class  RegSVDPolicy
 Implementation of the Regularized SVD policy to act as a wrapper when accessing Regularized SVD from within CFType. More...
 
class  SimilarityInterpolation
 With SimilarityInterpolation, interpolation weights are based on similarities between query user and its neighbors. More...
 
class  SVDCompletePolicy
 Implementation of the SVD complete incremental policy to act as a wrapper when accessing SVD complete decomposition from within CFType. More...
 
class  SVDIncompletePolicy
 Implementation of the SVD incomplete incremental to act as a wrapper when accessing SVD incomplete incremental from within CFType. More...
 
class  SVDPlusPlusPolicy
 Implementation of the SVDPlusPlus policy to act as a wrapper when accessing SVDPlusPlus from within CFType. More...
 
class  SVDWrapper
 This class acts as the wrapper for all SVD factorizers which are incompatible with CF module. More...
 
class  UserMeanNormalization
 This normalization class performs user mean normalization on raw ratings. More...
 
class  ZScoreNormalization
 This normalization class performs z-score normalization on raw ratings. More...
 

Typedefs

using EuclideanSearch = LMetricSearch< 2 >
 
typedef SVDWrapper< DummyClassArmaSVDFactorizer
 add simple typedefs
 

Enumerations

enum  NeighborSearchTypes { COSINE_SEARCH, EUCLIDEAN_SEARCH, PEARSON_SEARCH }
 NeighborSearchTypes contains the set of NeighborSearchPolicy classes that are usable by CFModel at prediction time.
 
enum  InterpolationTypes { AVERAGE_INTERPOLATION, REGRESSION_INTERPOLATION, SIMILARITY_INTERPOLATION }
 InterpolationTypes contains the set of InterpolationPolicy classes that are usable by CFModel at prediction time.
 

Functions

template<typename DecompositionPolicy >
CFWrapperBaseTrainHelper (const DecompositionPolicy &decomposition, const CFModel::NormalizationTypes normalizationType, const arma::mat &data, const size_t numUsersForSimilarity, const size_t rank, const size_t maxIterations, const double minResidue, const bool mit)
 
template<typename NeighborSearchPolicy , typename CFType >
void PredictHelper (CFType &cf, const InterpolationTypes interpolationType, const arma::Mat< size_t > &combinations, arma::vec &predictions)
 
template<typename NeighborSearchPolicy , typename CFType >
void GetRecommendationsHelper (CFType &cf, const InterpolationTypes interpolationType, const size_t numRecs, arma::Mat< size_t > &recommendations, const arma::Col< size_t > &users)
 
template<typename NeighborSearchPolicy , typename CFType >
void GetRecommendationsHelper (CFType &cf, const InterpolationTypes interpolationType, const size_t numRecs, arma::Mat< size_t > &recommendations)
 
template<typename DecompositionPolicy >
CFWrapperBaseInitializeModelHelper (CFModel::NormalizationTypes normalizationType)
 
CFWrapperBaseInitializeModel (CFModel::DecompositionTypes decompositionType, CFModel::NormalizationTypes normalizationType)
 
template<typename DecompositionPolicy , typename Archive >
void SerializeHelper (Archive &ar, CFWrapperBase *cf, CFModel::NormalizationTypes normalizationType)
 

Detailed Description

Collaborative filtering.