diff --git a/python/pyspark/mllib.py b/python/pyspark/mllib.py index e7e22166b016c88a030580d4e9b1c6a291033953..1877404cc13c006d6a2fb0d9c6d5ace8cabc0404 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