|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |
java.lang.Objectorg.mymedialite.ratingprediction.RatingPredictor
org.mymedialite.ratingprediction.IncrementalRatingPredictor
org.mymedialite.ratingprediction.KNN
org.mymedialite.ratingprediction.UserKNN
public abstract class UserKNN
Weighted user-based kNN.
Field Summary | |
---|---|
protected SparseBooleanMatrix |
data_user
boolean matrix indicating which user rated which item. |
Fields inherited from class org.mymedialite.ratingprediction.KNN |
---|
baseline_predictor, correlation, k |
Fields inherited from class org.mymedialite.ratingprediction.IncrementalRatingPredictor |
---|
updateItems, updateUsers |
Fields inherited from class org.mymedialite.ratingprediction.RatingPredictor |
---|
maxItemID, maxRating, maxUserID, minRating, ratings |
Constructor Summary | |
---|---|
UserKNN()
|
Method Summary | |
---|---|
void |
addRating(int user_id,
int item_id,
double rating)
Add a new rating and perform incremental training |
void |
addUser(int user_id)
|
int[] |
getMostSimilarUsers(int user_id,
int n)
get the most similar users. |
float |
getUserSimilarity(int user_id1,
int user_id2)
get the similarity between two users. |
double |
predict(int user_id,
int item_id)
Predict the rating of a given user for a given item. |
void |
removeRating(int user_id,
int item_id)
Remove an existing rating and perform "incremental" training |
protected abstract void |
retrainUser(int user_id)
Retrain model for a given user. |
void |
setRatings(IRatings ratings)
|
void |
updateRating(int user_id,
int item_id,
double rating)
Update an existing rating and perform incremental training |
Methods inherited from class org.mymedialite.ratingprediction.KNN |
---|
getRegI, getRegU, loadModel, saveModel, setRegI, setRegU |
Methods inherited from class org.mymedialite.ratingprediction.IncrementalRatingPredictor |
---|
addItem, getUpdateItems, getUpdateUsers, removeItem, removeUser, setUpdateItems, setUpdateUsers |
Methods inherited from class org.mymedialite.ratingprediction.RatingPredictor |
---|
canPredict, clone, getMaxRating, getMinRating, getRatings, setMaxRating, setMinRating, toString, train |
Methods inherited from class java.lang.Object |
---|
equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait |
Methods inherited from interface org.mymedialite.ratingprediction.IRatingPredictor |
---|
getMaxRating, getMinRating, setMaxRating, setMinRating |
Methods inherited from interface org.mymedialite.IRecommender |
---|
canPredict, toString, train |
Field Detail |
---|
protected SparseBooleanMatrix data_user
Constructor Detail |
---|
public UserKNN()
Method Detail |
---|
public void setRatings(IRatings ratings)
setRatings
in class KNN
public double predict(int user_id, int item_id)
predict
in interface IRecommender
predict
in class RatingPredictor
user_id
- the user IDitem_id
- the item ID
protected abstract void retrainUser(int user_id)
user_id
- the user IDpublic void addRating(int user_id, int item_id, double rating)
IIncrementalRatingPredictor
addRating
in interface IIncrementalRatingPredictor
addRating
in class IncrementalRatingPredictor
user_id
- the ID of the user who performed the ratingitem_id
- the ID of the rated itemrating
- the rating valuepublic void updateRating(int user_id, int item_id, double rating)
IIncrementalRatingPredictor
updateRating
in interface IIncrementalRatingPredictor
updateRating
in class IncrementalRatingPredictor
user_id
- the ID of the user who performed the ratingitem_id
- the ID of the rated itemrating
- the rating valuepublic void removeRating(int user_id, int item_id)
IIncrementalRatingPredictor
removeRating
in interface IIncrementalRatingPredictor
removeRating
in class IncrementalRatingPredictor
user_id
- the ID of the user who performed the ratingitem_id
- the ID of the rated itempublic void addUser(int user_id)
addUser
in class IncrementalRatingPredictor
public float getUserSimilarity(int user_id1, int user_id2)
IUserSimilarityProvider
getUserSimilarity
in interface IUserSimilarityProvider
user_id1
- the ID of the first useruser_id2
- the ID of the second user
public int[] getMostSimilarUsers(int user_id, int n)
IUserSimilarityProvider
getMostSimilarUsers
in interface IUserSimilarityProvider
user_id
- the ID of the usern
- the number of similar users to return
|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |