diff options
| author | MechCoder <manojkumarsivaraj334@gmail.com> | 2015-04-30 23:51:00 -0700 |
|---|---|---|
| committer | Xiangrui Meng <meng@databricks.com> | 2015-04-30 23:51:00 -0700 |
| commit | c24aeb6a310b49dba8db1f4642531780a2e27253 (patch) | |
| tree | 78a78ef9682b92eb57bccb91c445fefaf9bb238e /docs/mllib-collaborative-filtering.md | |
| parent | 14b32886fa01aef6cc0dfbc263eb6d4c9d2876fa (diff) | |
| download | spark-c24aeb6a310b49dba8db1f4642531780a2e27253.tar.gz spark-c24aeb6a310b49dba8db1f4642531780a2e27253.tar.bz2 spark-c24aeb6a310b49dba8db1f4642531780a2e27253.zip | |
[SPARK-6257] [PYSPARK] [MLLIB] MLlib API missing items in Recommendation
Adds
rank, recommendUsers and RecommendProducts to MatrixFactorizationModel in PySpark.
Author: MechCoder <manojkumarsivaraj334@gmail.com>
Closes #5807 from MechCoder/spark-6257 and squashes the following commits:
09629c6 [MechCoder] doc
953b326 [MechCoder] [SPARK-6257] MLlib API missing items in Recommendation
Diffstat (limited to 'docs/mllib-collaborative-filtering.md')
| -rw-r--r-- | docs/mllib-collaborative-filtering.md | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/docs/mllib-collaborative-filtering.md b/docs/mllib-collaborative-filtering.md index 76140282a2..7b397e30b2 100644 --- a/docs/mllib-collaborative-filtering.md +++ b/docs/mllib-collaborative-filtering.md @@ -216,7 +216,7 @@ model = ALS.train(ratings, rank, numIterations) testdata = ratings.map(lambda p: (p[0], p[1])) predictions = model.predictAll(testdata).map(lambda r: ((r[0], r[1]), r[2])) ratesAndPreds = ratings.map(lambda r: ((r[0], r[1]), r[2])).join(predictions) -MSE = ratesAndPreds.map(lambda r: (r[1][0] - r[1][1])**2).reduce(lambda x, y: x + y) / ratesAndPreds.count() +MSE = ratesAndPreds.map(lambda r: (r[1][0] - r[1][1])**2).mean() print("Mean Squared Error = " + str(MSE)) # Save and load model |