Skip to content

Commit 4b8dc25

Browse files
Viveka Kulhariasrowen
Viveka Kulharia
authored andcommitted
rmse was wrongly calculated
It was multiplying with U instaed of dividing by U Author: Viveka Kulharia <vivkul@iitk.ac.in> Closes #9771 from vivkul/patch-1. (cherry picked from commit 1429e0a) Signed-off-by: Sean Owen <sowen@cloudera.com>
1 parent 0ed6d9c commit 4b8dc25

File tree

1 file changed

+1
-1
lines changed
  • examples/src/main/python

1 file changed

+1
-1
lines changed

examples/src/main/python/als.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636

3737
def rmse(R, ms, us):
3838
diff = R - ms * us.T
39-
return np.sqrt(np.sum(np.power(diff, 2)) / M * U)
39+
return np.sqrt(np.sum(np.power(diff, 2)) / (M * U))
4040

4141

4242
def update(i, vec, mat, ratings):

0 commit comments

Comments
 (0)