Skip to content

Commit dd71754

Browse files
committed
SGD update with sigmoid
1 parent 528f934 commit dd71754

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,14 @@
11
package hu.sztaki.ilab.ps.matrix.factorization.factors
22

3+
import breeze.numerics.sigmoid
4+
35
class SGDUpdater(learningRate: Double) extends FactorUpdater {
46

57
override def delta(rating: Double, user: Array[Double], item: Array[Double]): (Array[Double], Array[Double]) = {
6-
val e = rating - user.zip(item).map { case (x, y) => x * y }.sum
7-
val userItem = user.zip(item)
8+
val error = sigmoid(rating - user.zip(item).map { case (x, y) => x * y }.sum)
89

9-
(item.map(i => learningRate * e * i),
10-
user.map(u => learningRate * e * u))
10+
(item.map(i => learningRate * error * i),
11+
user.map(u => learningRate * error * u))
1112
}
1213

1314
}

0 commit comments

Comments
 (0)