mlpack
|
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< DummyClass > | ArmaSVDFactorizer |
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 > | |
CFWrapperBase * | TrainHelper (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 > | |
CFWrapperBase * | InitializeModelHelper (CFModel::NormalizationTypes normalizationType) |
CFWrapperBase * | InitializeModel (CFModel::DecompositionTypes decompositionType, CFModel::NormalizationTypes normalizationType) |
template<typename DecompositionPolicy , typename Archive > | |
void | SerializeHelper (Archive &ar, CFWrapperBase *cf, CFModel::NormalizationTypes normalizationType) |
Collaborative filtering.