diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/pyspark/mllib.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/pyspark/mllib.py b/python/pyspark/mllib.py index e7e22166b0..1877404cc1 100644 --- a/python/pyspark/mllib.py +++ b/python/pyspark/mllib.py @@ -276,7 +276,7 @@ class SVMModel(LinearModel): class KMeansModel(object): """A clustering model derived from the k-means method. - >>> data = array([0.0, 0.0, 1.0,1.0, 9.0,8.0, 8.0,9.0]).reshape(4,2) + >>> data = array([0.0,0.0, 1.0,1.0, 9.0,8.0, 8.0,9.0]).reshape(4,2) >>> clusters = KMeansModel.train(sc, sc.parallelize(data), 2, maxIterations=10, runs=30, initialization_mode="random") >>> clusters.predict(array([0.0, 0.0])) == clusters.predict(array([1.0, 1.0])) True |