org.lenskit.mf.svd

Class MFModel

java.lang.Object

org.lenskit.mf.svd.MFModel

All Implemented Interfaces:

Serializable

Direct Known Subclasses:

FunkSVDModel

public class MFModel
extends Object
implements Serializable

Common model for matrix factorization (SVD) recommendation.

Since:

2.1

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected int	featureCount
protected int	itemCount
protected KeyIndex	itemIndex
protected org.apache.commons.math3.linear.RealMatrix	itemMatrix
protected int	userCount
protected KeyIndex	userIndex
protected org.apache.commons.math3.linear.RealMatrix	userMatrix

Constructor Summary

Constructors

Constructor and Description
MFModel(org.apache.commons.math3.linear.RealMatrix umat, org.apache.commons.math3.linear.RealMatrix imat, KeyIndex uidx, KeyIndex iidx) Construct a matrix factorization model.

Method Summary

Modifier and Type	Method and Description
int	getFeatureCount() Get the model’s feature count.
int	getItemCount() The number of items.
double	getItemFeature(long iid, int feature) Get a particular feature value for an item.
KeyIndex	getItemIndex() The item index mapping.
org.apache.commons.math3.linear.RealMatrix	getItemMatrix() Get the item matrix.
org.apache.commons.math3.linear.RealVector	getItemVector(long item)
int	getUserCount() The number of users.
double	getUserFeature(long uid, int feature) Get a particular feature value for an user.
KeyIndex	getUserIndex() The user index mapping.
org.apache.commons.math3.linear.RealMatrix	getUserMatrix() Get the user matrix.
org.apache.commons.math3.linear.RealVector	getUserVector(long user)
String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Field Detail

featureCount

protected int featureCount

userCount

protected int userCount

itemCount

protected int itemCount

userMatrix

protected org.apache.commons.math3.linear.RealMatrix userMatrix

itemMatrix

protected org.apache.commons.math3.linear.RealMatrix itemMatrix

userIndex

protected KeyIndex userIndex

itemIndex

protected KeyIndex itemIndex

Constructor Detail

MFModel

public MFModel(org.apache.commons.math3.linear.RealMatrix umat,
               org.apache.commons.math3.linear.RealMatrix imat,
               KeyIndex uidx,
               KeyIndex iidx)

Construct a matrix factorization model. The matrices are not copied, so the caller should make sure they won’t be modified by anyone else.

Parameters:

umat - The user feature matrix (users x features).

imat - The item feature matrix (items x features).

uidx - The user index mapping.

iidx - The item index mapping.

Method Detail

getFeatureCount

public int getFeatureCount()

Get the model’s feature count.

Returns:

The model’s feature count.

getUserCount

public int getUserCount()

The number of users.

getItemCount

public int getItemCount()

The number of items.

getItemIndex

public KeyIndex getItemIndex()

The item index mapping.

getUserIndex

public KeyIndex getUserIndex()

The user index mapping.

getUserMatrix

public org.apache.commons.math3.linear.RealMatrix getUserMatrix()

Get the user matrix.

Returns:

The user matrix (users x features).

getItemMatrix

public org.apache.commons.math3.linear.RealMatrix getItemMatrix()

Get the item matrix.

Returns:

The item matrix (items x features).

getUserVector

@Nullable
public org.apache.commons.math3.linear.RealVector getUserVector(long user)

getItemVector

@Nullable
public org.apache.commons.math3.linear.RealVector getItemVector(long item)

getUserFeature

public double getUserFeature(long uid,
                             int feature)

Get a particular feature value for an user.

Parameters:

uid - The item ID.

feature - The feature.

Returns:

The user-feature value, or 0 if the user was not in the training set.

getItemFeature

public double getItemFeature(long iid,
                             int feature)

Get a particular feature value for an item.

Parameters:

iid - The item ID.

feature - The feature.

Returns:

The item-feature value, or 0 if the item was not in the training set.

toString

public String toString()

Overrides:

toString in class Object