diff options
author | WeichenXu <WeichenXu123@outlook.com> | 2016-07-24 02:29:08 -0700 |
---|---|---|
committer | Yanbo Liang <ybliang8@gmail.com> | 2016-07-24 02:29:08 -0700 |
commit | 37bed97de5f81a9127f1ff12db695aab6e5c4c47 (patch) | |
tree | e7a6e13aec75e037f814372a8ad6af6579232ba6 | |
parent | cc1d2dcb612fb5df39c9a9e57a3484ecad90c745 (diff) | |
download | spark-37bed97de5f81a9127f1ff12db695aab6e5c4c47.tar.gz spark-37bed97de5f81a9127f1ff12db695aab6e5c4c47.tar.bz2 spark-37bed97de5f81a9127f1ff12db695aab6e5c4c47.zip |
[PYSPARK] add picklable SparseMatrix in pyspark.ml.common
## What changes were proposed in this pull request?
add `SparseMatrix` class whick support pickler.
## How was this patch tested?
Existing test.
Author: WeichenXu <WeichenXu123@outlook.com>
Closes #14265 from WeichenXu123/picklable_py.
-rw-r--r-- | python/pyspark/ml/common.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/python/pyspark/ml/common.py b/python/pyspark/ml/common.py index 7d449aaccb..aec860fca7 100644 --- a/python/pyspark/ml/common.py +++ b/python/pyspark/ml/common.py @@ -51,6 +51,7 @@ py4j.protocol.smart_decode = _new_smart_decode _picklable_classes = [ 'SparseVector', 'DenseVector', + 'SparseMatrix', 'DenseMatrix', ] |