public final class SVDPlusPlusFactorizer extends RatingSGDFactorizer
biasLearningRate, biasReg, dataModel, FEATURE_OFFSET, ITEM_BIAS_INDEX, itemVectors, learningRate, learningRateDecay, numFeatures, preventOverfitting, randomNoise, USER_BIAS_INDEX, userVectors
Constructor and Description |
---|
SVDPlusPlusFactorizer(DataModel dataModel,
int numFeatures,
double learningRate,
double preventOverfitting,
double randomNoise,
int numIterations,
double learningRateDecay) |
SVDPlusPlusFactorizer(DataModel dataModel,
int numFeatures,
int numIterations) |
Modifier and Type | Method and Description |
---|---|
Factorization |
factorize() |
protected void |
prepareTraining() |
protected void |
updateParameters(long userID,
long itemID,
float rating,
double currentLearningRate) |
shufflePreferences
createFactorization, itemIndex, refresh, userIndex
public SVDPlusPlusFactorizer(DataModel dataModel, int numFeatures, int numIterations) throws TasteException
TasteException
public SVDPlusPlusFactorizer(DataModel dataModel, int numFeatures, double learningRate, double preventOverfitting, double randomNoise, int numIterations, double learningRateDecay) throws TasteException
TasteException
protected void prepareTraining() throws TasteException
prepareTraining
in class RatingSGDFactorizer
TasteException
public Factorization factorize() throws TasteException
factorize
in interface Factorizer
factorize
in class RatingSGDFactorizer
TasteException
protected void updateParameters(long userID, long itemID, float rating, double currentLearningRate)
updateParameters
in class RatingSGDFactorizer
Copyright © 2008–2017 The Apache Software Foundation. All rights reserved.