-
- Downloads
[SPARK-19535][ML] RecommendForAllUsers RecommendForAllItems for ALS on Dataframe
## What changes were proposed in this pull request? This is a simple implementation of RecommendForAllUsers & RecommendForAllItems for the Dataframe version of ALS. It uses Dataframe operations (not a wrapper on the RDD implementation). Haven't benchmarked against a wrapper, but unit test examples do work. ## How was this patch tested? Unit tests ``` $ build/sbt > mllib/testOnly *ALSSuite -- -z "recommendFor" > mllib/testOnly ``` Author: Your Name <you@example.com> Author: sueann <sueann@databricks.com> Closes #17090 from sueann/SPARK-19535.
Showing
- mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala 70 additions, 9 deletions...c/main/scala/org/apache/spark/ml/recommendation/ALS.scala
- mllib/src/main/scala/org/apache/spark/ml/recommendation/TopByKeyAggregator.scala 60 additions, 0 deletions...g/apache/spark/ml/recommendation/TopByKeyAggregator.scala
- mllib/src/test/scala/org/apache/spark/ml/recommendation/ALSSuite.scala 94 additions, 0 deletions...t/scala/org/apache/spark/ml/recommendation/ALSSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/recommendation/TopByKeyAggregatorSuite.scala 73 additions, 0 deletions...che/spark/ml/recommendation/TopByKeyAggregatorSuite.scala
Loading
Please register or sign in to comment