Skip to content
Snippets Groups Projects
Commit 1429e0a2 authored by Viveka Kulharia's avatar Viveka Kulharia Committed by Sean Owen
Browse files

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.
parent 9631ca35
No related branches found
No related tags found
No related merge requests found
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment