diff options
author | Xiangrui Meng <meng@databricks.com> | 2015-03-17 12:14:40 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-03-17 12:14:40 -0700 |
commit | c94d0626471e209ab7ebfc588f9a2992946b7ed5 (patch) | |
tree | b5025c4412aac661e915fe25f3f0a9aaaa371a50 /python/pyspark/mllib/common.py | |
parent | d9f3e01688ad0a8d5fc2419a948a682ad7d957c9 (diff) | |
download | spark-c94d0626471e209ab7ebfc588f9a2992946b7ed5.tar.gz spark-c94d0626471e209ab7ebfc588f9a2992946b7ed5.tar.bz2 spark-c94d0626471e209ab7ebfc588f9a2992946b7ed5.zip |
[SPARK-6226][MLLIB] add save/load in PySpark's KMeansModel
Use `_py2java` and `_java2py` to convert Python model to/from Java model. yinxusen
Author: Xiangrui Meng <meng@databricks.com>
Closes #5049 from mengxr/SPARK-6226-mengxr and squashes the following commits:
570ba81 [Xiangrui Meng] fix python style
b10b911 [Xiangrui Meng] add save/load in PySpark's KMeansModel
Diffstat (limited to 'python/pyspark/mllib/common.py')
-rw-r--r-- | python/pyspark/mllib/common.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/python/pyspark/mllib/common.py b/python/pyspark/mllib/common.py index 621591c26b..a539d2f284 100644 --- a/python/pyspark/mllib/common.py +++ b/python/pyspark/mllib/common.py @@ -70,8 +70,8 @@ def _py2java(sc, obj): obj = _to_java_object_rdd(obj) elif isinstance(obj, SparkContext): obj = obj._jsc - elif isinstance(obj, list) and (obj or isinstance(obj[0], JavaObject)): - obj = ListConverter().convert(obj, sc._gateway._gateway_client) + elif isinstance(obj, list): + obj = ListConverter().convert([_py2java(sc, x) for x in obj], sc._gateway._gateway_client) elif isinstance(obj, JavaObject): pass elif isinstance(obj, (int, long, float, bool, basestring)): |