CFModel Class Reference

The model to save to disk. More...

Public Member Functions

 CFModel ()
 Create an empty CF model. More...

 
 ~CFModel ()
 Clean up memory. More...

 
template
<
typename
DecompositionPolicy
,
typename
NormalizationType
=
NoNormalization
>
const CFType< DecompositionPolicy, NormalizationType > * CFPtr () const
 Get the pointer to CFType<> object. More...

 
template
<
typename
NeighborSearchPolicy
,
typename
InterpolationPolicy
>
void GetRecommendations (const size_t numRecs, arma::Mat< size_t > &recommendations, const arma::Col< size_t > &users)
 Compute recommendations for query users. More...

 
template
<
typename
NeighborSearchPolicy
,
typename
InterpolationPolicy
>
void GetRecommendations (const size_t numRecs, arma::Mat< size_t > &recommendations)
 Compute recommendations for all users. More...

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

 
template
<
typename
Archive
>
void serialize (Archive &ar, const unsigned int)
 Serialize the model. More...

 
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, const std::string &normalizationType="none")
 Train the model. More...

 

Detailed Description

The model to save to disk.

Definition at line 121 of file cf_model.hpp.

Constructor & Destructor Documentation

◆ CFModel()

CFModel ( )
inline

Create an empty CF model.

Definition at line 177 of file cf_model.hpp.

◆ ~CFModel()

~CFModel ( )

Clean up memory.

Member Function Documentation

◆ CFPtr()

const CFType<DecompositionPolicy, NormalizationType>* CFPtr ( ) const

Get the pointer to CFType<> object.

◆ GetRecommendations() [1/2]

void GetRecommendations ( const size_t  numRecs,
arma::Mat< size_t > &  recommendations,
const arma::Col< size_t > &  users 
)

Compute recommendations for query users.

◆ GetRecommendations() [2/2]

void GetRecommendations ( const size_t  numRecs,
arma::Mat< size_t > &  recommendations 
)

Compute recommendations for all users.

◆ Predict()

void Predict ( const arma::Mat< size_t > &  combinations,
arma::vec &  predictions 
)

Make predictions.

◆ serialize()

void serialize ( Archive &  ar,
const unsigned  int 
)

Serialize the model.

◆ Train()

void Train ( const MatType &  data,
const size_t  numUsersForSimilarity,
const size_t  rank,
const size_t  maxIterations,
const double  minResidue,
const bool  mit,
const std::string &  normalizationType = "none" 
)

Train the model.


The documentation for this class was generated from the following file:
  • /home/ryan/src/mlpack.org-go/_src/mlpack-git/src/mlpack/methods/cf/cf_model.hpp