-
- Downloads
[SPARK-14412][ML][PYSPARK] Add StorageLevel params to ALS
`mllib` `ALS` supports `setIntermediateRDDStorageLevel` and `setFinalRDDStorageLevel`. This PR adds these as Params in `ml` `ALS`. They are put in group **expertParam** since few users will need them. ## How was this patch tested? New test cases in `ALSSuite` and `tests.py`. cc yanboliang jkbradley sethah rishabhbhardwaj Author: Nick Pentreath <nickp@za.ibm.com> Closes #12660 from MLnick/SPARK-14412-als-storage-params.
Showing
- mllib/src/main/scala/org/apache/spark/ml/recommendation/ALS.scala 49 additions, 5 deletions...c/main/scala/org/apache/spark/ml/recommendation/ALS.scala
- mllib/src/test/scala/org/apache/spark/ml/recommendation/ALSSuite.scala 80 additions, 1 deletion...t/scala/org/apache/spark/ml/recommendation/ALSSuite.scala
- python/pyspark/ml/recommendation.py 53 additions, 5 deletionspython/pyspark/ml/recommendation.py
- python/pyspark/ml/tests.py 27 additions, 0 deletionspython/pyspark/ml/tests.py
Loading
Please register or sign in to comment