Simple matrix factorization class. More...
Public Member Functions | |
override void | AddRating (int user_id, int item_id, double rating) |
virtual bool | CanPredict (int user_id, int item_id) |
Check whether a useful prediction can be made for a given user-item combination. | |
Object | Clone () |
create a shallow copy of the object | |
double | ComputeFit () |
Compute the fit (RMSE) on the training data. | |
virtual double | ComputeLoss () |
Compute the regularized loss. | |
virtual void | Iterate () |
Run one iteration (= pass over the training data). | |
override void | LoadModel (string filename) |
Get the model parameters from a file. | |
MatrixFactorization () | |
Default constructor. | |
override double | Predict (int user_id, int item_id) |
Predict the rating of a given user for a given item. | |
override void | RemoveItem (int item_id) |
override void | RemoveRating (int user_id, int item_id) |
override void | RemoveUser (int user_id) |
virtual void | RetrainItem (int item_id) |
Updates the latent factors of an item. | |
virtual void | RetrainUser (int user_id) |
Updates the latent factors on a user. | |
override void | SaveModel (string filename) |
Save the model parameters to a file. | |
override string | ToString () |
Return a string representation of the recommender. | |
override void | Train () |
Learn the model parameters of the recommender from the training data. | |
override void | UpdateRating (int user_id, int item_id, double rating) |
Protected Member Functions | |
override void | AddItem (int item_id) |
override void | AddUser (int user_id) |
override void | InitModel () |
Initialize the model data structure. | |
virtual void | Iterate (IList< int > rating_indices, bool update_user, bool update_item) |
Iterate once over rating data and adjust corresponding factors (stochastic gradient descent). | |
double | Predict (int user_id, int item_id, bool bound) |
Protected Attributes | |
double | global_bias |
The bias (global average). | |
Matrix< double > | item_factors |
Matrix containing the latent item factors. | |
double | max_rating |
The max rating value. | |
double | min_rating |
The min rating value. | |
IRatings | ratings |
rating data | |
Matrix< double > | user_factors |
Matrix containing the latent user factors. | |
Properties | |
double | InitMean [get, set] |
Mean of the normal distribution used to initialize the factors. | |
double | InitStdev [get, set] |
Standard deviation of the normal distribution used to initialize the factors. | |
double | LearnRate [get, set] |
Learn rate. | |
int | MaxItemID [get, set] |
Maximum item ID. | |
virtual double | MaxRating [get, set] |
The max rating value. | |
int | MaxUserID [get, set] |
Maximum user ID. | |
virtual double | MinRating [get, set] |
The min rating value. | |
uint | NumFactors [get, set] |
Number of latent factors. | |
uint | NumIter [get, set] |
Number of iterations over the training data. | |
virtual IRatings | Ratings [get, set] |
The rating data. | |
virtual double | Regularization [get, set] |
Regularization parameter. | |
bool | UpdateItems [get, set] |
true if items shall be updated when doing incremental updates | |
bool | UpdateUsers [get, set] |
true if users shall be updated when doing incremental updates |
Simple matrix factorization class.
Factorizing the observed rating values using a factor matrix for users and one for items.
NaN values in the model occur if values become too large or too small to be represented by the type double. If you encounter such problems, there are three ways to fix them: (1) (preferred) Use BiasedMatrixFactorization, which is more stable. (2) Change the range of rating values (1 to 5 works generally well with the default settings). (3) Change the learn_rate (decrease it if your range is larger than 1 to 5).
This recommender supports incremental updates.
Default constructor.
virtual bool CanPredict | ( | int | user_id, | |
int | item_id | |||
) | [virtual, inherited] |
Check whether a useful prediction can be made for a given user-item combination.
user_id | the user ID | |
item_id | the item ID |
Implements IRecommender.
Reimplemented in BiPolarSlopeOne, GlobalAverage, ItemAverage, SlopeOne, and UserAverage.
Object Clone | ( | ) | [inherited] |
create a shallow copy of the object
double ComputeFit | ( | ) |
Compute the fit (RMSE) on the training data.
Implements IIterativeModel.
virtual double ComputeLoss | ( | ) | [virtual] |
Compute the regularized loss.
Reimplemented in BiasedMatrixFactorization, and BiasedMatrixFactorizationMAE.
override void InitModel | ( | ) | [protected, virtual] |
Initialize the model data structure.
Reimplemented from RatingPredictor.
Reimplemented in BiasedMatrixFactorization.
virtual void Iterate | ( | IList< int > | rating_indices, | |
bool | update_user, | |||
bool | update_item | |||
) | [protected, virtual] |
Iterate once over rating data and adjust corresponding factors (stochastic gradient descent).
rating_indices | a list of indices pointing to the ratings to iterate over | |
update_user | true if user factors to be updated | |
update_item | true if item factors to be updated |
Reimplemented in BiasedMatrixFactorization, and BiasedMatrixFactorizationMAE.
virtual void Iterate | ( | ) | [virtual] |
Run one iteration (= pass over the training data).
Implements IIterativeModel.
Reimplemented in BiasedMatrixFactorization.
override void LoadModel | ( | string | filename | ) | [virtual] |
Get the model parameters from a file.
filename | the name of the file to read from |
Implements RatingPredictor.
Reimplemented in BiasedMatrixFactorization.
override double Predict | ( | int | user_id, | |
int | item_id | |||
) | [virtual] |
Predict the rating of a given user for a given item.
If the user or the item are not known to the recommender, the global average is returned. To avoid this behavior for unknown entities, use CanPredict() to check before.
user_id | the user ID | |
item_id | the item ID |
Implements RatingPredictor.
Reimplemented in BiasedMatrixFactorization.
virtual void RetrainItem | ( | int | item_id | ) | [virtual] |
Updates the latent factors of an item.
item_id | the item ID |
Reimplemented in BiasedMatrixFactorization.
virtual void RetrainUser | ( | int | user_id | ) | [virtual] |
Updates the latent factors on a user.
user_id | the user ID |
Reimplemented in BiasedMatrixFactorization.
override void SaveModel | ( | string | filename | ) | [virtual] |
Save the model parameters to a file.
filename | the name of the file to write to |
Implements RatingPredictor.
Reimplemented in BiasedMatrixFactorization.
override string ToString | ( | ) |
Return a string representation of the recommender.
The ToString() method of recommenders should list the class name and all hyperparameters, separated by space characters.
Implements IRecommender.
Reimplemented in BiasedMatrixFactorization, and BiasedMatrixFactorizationMAE.
double global_bias [protected] |
The bias (global average).
Matrix<double> item_factors [protected] |
Matrix containing the latent item factors.
double max_rating [protected, inherited] |
The max rating value.
double min_rating [protected, inherited] |
The min rating value.
Matrix<double> user_factors [protected] |
Matrix containing the latent user factors.
double InitMean [get, set] |
Mean of the normal distribution used to initialize the factors.
double InitStdev [get, set] |
Standard deviation of the normal distribution used to initialize the factors.
double LearnRate [get, set] |
Learn rate.
int MaxItemID [get, set, inherited] |
Maximum item ID.
virtual double MaxRating [get, set, inherited] |
The max rating value.
Implements IRatingPredictor.
int MaxUserID [get, set, inherited] |
Maximum user ID.
virtual double MinRating [get, set, inherited] |
The min rating value.
Implements IRatingPredictor.
uint NumFactors [get, set] |
Number of latent factors.
uint NumIter [get, set] |
Number of iterations over the training data.
Implements IIterativeModel.
virtual double Regularization [get, set] |
Regularization parameter.
Reimplemented in BiasedMatrixFactorization.
bool UpdateItems [get, set, inherited] |
true if items shall be updated when doing incremental updates
Default is true. Set to false if you do not want any updates to the item model parameters when doing incremental updates.
bool UpdateUsers [get, set, inherited] |
true if users shall be updated when doing incremental updates
Default is true. Set to false if you do not want any updates to the user model parameters when doing incremental updates.