-
- Downloads
[SPARK-4531] [MLlib] cache serialized java object
The Pyrolite is pretty slow (comparing to the adhoc serializer in 1.1), it cause much performance regression in 1.2, because we cache the serialized Python object in JVM, deserialize them into Java object in each step. This PR change to cache the deserialized JavaRDD instead of PythonRDD to avoid the deserialization of Pyrolite. It should have similar memory usage as before, but much faster. Author: Davies Liu <davies@databricks.com> Closes #3397 from davies/cache and squashes the following commits: 7f6e6ce [Davies Liu] Update -> Updater 4b52edd [Davies Liu] using named argument 63b984e [Davies Liu] fix 7da0332 [Davies Liu] add unpersist() dff33e1 [Davies Liu] address comments c2bdfc2 [Davies Liu] refactor d572f00 [Davies Liu] Merge branch 'master' into cache f1063e1 [Davies Liu] cache serialized java object
Showing
- mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala 52 additions, 58 deletions...la/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala
- mllib/src/main/scala/org/apache/spark/mllib/clustering/KMeans.scala 2 additions, 11 deletions...main/scala/org/apache/spark/mllib/clustering/KMeans.scala
- mllib/src/main/scala/org/apache/spark/mllib/regression/GeneralizedLinearAlgorithm.scala 2 additions, 11 deletions...e/spark/mllib/regression/GeneralizedLinearAlgorithm.scala
- python/pyspark/mllib/clustering.py 3 additions, 5 deletionspython/pyspark/mllib/clustering.py
- python/pyspark/mllib/common.py 1 addition, 3 deletionspython/pyspark/mllib/common.py
- python/pyspark/mllib/recommendation.py 2 additions, 2 deletionspython/pyspark/mllib/recommendation.py
- python/pyspark/mllib/regression.py 2 additions, 3 deletionspython/pyspark/mllib/regression.py
Loading
Please register or sign in to comment