From 1429e0a2b562469146b6fa06051c85a00092e5b8 Mon Sep 17 00:00:00 2001
From: Viveka Kulharia <vivkul@iitk.ac.in>
Date: Wed, 18 Nov 2015 09:10:15 +0000
Subject: [PATCH] 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.
---
examples/src/main/python/als.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/examples/src/main/python/als.py b/examples/src/main/python/als.py
index 1c3a787bd0..205ca02962 100755
--- a/examples/src/main/python/als.py
+++ b/examples/src/main/python/als.py
@@ -36,7 +36,7 @@ np.random.seed(42)
def rmse(R, ms, us):
diff = R - ms * us.T
- return np.sqrt(np.sum(np.power(diff, 2)) / M * U)
+ return np.sqrt(np.sum(np.power(diff, 2)) / (M * U))
def update(i, vec, mat, ratings):
--
GitLab