|
|||||||||
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.ItemKNN
public abstract class ItemKNN
Weighted item-based kNN.
Field Summary | |
---|---|
protected SparseBooleanMatrix |
data_item
Matrix indicating which item was rated by which user. |
protected Memoizer<java.lang.Integer,java.util.List<java.lang.Integer>> |
memoizer
Source of positively correlated entities. |
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 | |
---|---|
ItemKNN()
|
Method Summary | |
---|---|
void |
addItem(int item_id)
|
void |
addRating(int user_id,
int item_id,
double rating)
Add a new rating and perform incremental training |
float |
getItemSimilarity(int item_id1,
int item_id2)
Get the similarity between two items. |
int[] |
getMostSimilarItems(int item_id,
int n)
Get the most similar items. |
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 |
retrainItem(int item_id)
Retrain model for a given item. |
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 |
---|
addUser, 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_item
protected Memoizer<java.lang.Integer,java.util.List<java.lang.Integer>> memoizer
Constructor Detail |
---|
public ItemKNN()
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 retrainItem(int item_id)
item_id
- the item 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 addItem(int item_id)
addItem
in class IncrementalRatingPredictor
public float getItemSimilarity(int item_id1, int item_id2)
IItemSimilarityProvider
getItemSimilarity
in interface IItemSimilarityProvider
item_id1
- the ID of the first itemitem_id2
- the ID of the second item
public int[] getMostSimilarItems(int item_id, int n)
IItemSimilarityProvider
getMostSimilarItems
in interface IItemSimilarityProvider
item_id
- the ID of the itemn
- the number of similar items to return
|
|||||||||
PREV CLASS NEXT CLASS | FRAMES NO FRAMES | ||||||||
SUMMARY: NESTED | FIELD | CONSTR | METHOD | DETAIL: FIELD | CONSTR | METHOD |