diff --git a/core/src/main/java/net/librec/recommender/cf/ranking/FISMrmseRecommender.java b/core/src/main/java/net/librec/recommender/cf/ranking/FISMrmseRecommender.java index 56fc0478e..190102560 100644 --- a/core/src/main/java/net/librec/recommender/cf/ranking/FISMrmseRecommender.java +++ b/core/src/main/java/net/librec/recommender/cf/ranking/FISMrmseRecommender.java @@ -161,7 +161,7 @@ protected void trainModel() throws LibrecException { loss += itemBiasReg * bi * bi; // update bu userBiases.plus(u, lRate * (eui - userBiasReg * bu)); - loss += itemBiasReg * bu * bu; + loss += userBiasReg * bu * bu; // update Qi DenseVector deltaq = X.times(eui).minus(Q.row(i).times(beta)); loss += beta * Q.row(i).dot(Q.row(i));