prea.recommender.baseline
Class Average
java.lang.Object
prea.recommender.baseline.BaselineRecommender
prea.recommender.baseline.Average
- All Implemented Interfaces:
- Recommender
public class Average
- extends BaselineRecommender
The class implementing a baseline, predicting by overall average of training set ratings.
- Since:
- 2012. 4. 20
- Version:
- 1.1
- Author:
- Joonseok Lee
|
Field Summary |
private double |
constantValue
The value which will be used for predicting all ratings. |
|
Constructor Summary |
Average(int uc,
int ic,
double max,
double min)
Construct a constant model with the given data. |
|
Method Summary |
void |
buildModel(SparseMatrix rm)
Build a model with given training set. |
double |
predict(int userId,
int itemId)
Predict a rating for the given user and item. |
| Methods inherited from class java.lang.Object |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
constantValue
private double constantValue
- The value which will be used for predicting all ratings.
Average
public Average(int uc,
int ic,
double max,
double min)
- Construct a constant model with the given data.
- Parameters:
uc - The number of users in the dataset.ic - The number of items in the dataset.max - The maximum rating value in the dataset.min - The minimum rating value in the dataset.
buildModel
public void buildModel(SparseMatrix rm)
- Build a model with given training set.
- Specified by:
buildModel in interface Recommender- Overrides:
buildModel in class BaselineRecommender
- Parameters:
rm - Training data set.
predict
public double predict(int userId,
int itemId)
- Predict a rating for the given user and item.
- Specified by:
predict in class BaselineRecommender
- Parameters:
userId - The target user.itemId - The target item.
- Returns:
- predicted rating.