Skip to content

Commit 7cf662b

Browse files
committed
[SPARK-24973][PYTHON]Add numIter to Python ClusteringSummary
1 parent d6b7545 commit 7cf662b

File tree

2 files changed

+11
-0
lines changed

2 files changed

+11
-0
lines changed

python/pyspark/ml/clustering.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,14 @@ def clusterSizes(self):
8888
"""
8989
return self._call_java("clusterSizes")
9090

91+
@property
92+
@since("2.4.0")
93+
def numIter(self):
94+
"""
95+
Number of iterations.
96+
"""
97+
return self._call_java("numIter")
98+
9199

92100
class GaussianMixtureModel(JavaModel, JavaMLWritable, JavaMLReadable):
93101
"""

python/pyspark/ml/tests.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1888,6 +1888,7 @@ def test_gaussian_mixture_summary(self):
18881888
self.assertTrue(isinstance(s.cluster, DataFrame))
18891889
self.assertEqual(len(s.clusterSizes), 2)
18901890
self.assertEqual(s.k, 2)
1891+
self.assertEqual(s.numIter, 3)
18911892

18921893
def test_bisecting_kmeans_summary(self):
18931894
data = [(Vectors.dense(1.0),), (Vectors.dense(5.0),), (Vectors.dense(10.0),),
@@ -1903,6 +1904,7 @@ def test_bisecting_kmeans_summary(self):
19031904
self.assertTrue(isinstance(s.cluster, DataFrame))
19041905
self.assertEqual(len(s.clusterSizes), 2)
19051906
self.assertEqual(s.k, 2)
1907+
self.assertEqual(s.numIter, 20)
19061908

19071909
def test_kmeans_summary(self):
19081910
data = [(Vectors.dense([0.0, 0.0]),), (Vectors.dense([1.0, 1.0]),),
@@ -1918,6 +1920,7 @@ def test_kmeans_summary(self):
19181920
self.assertTrue(isinstance(s.cluster, DataFrame))
19191921
self.assertEqual(len(s.clusterSizes), 2)
19201922
self.assertEqual(s.k, 2)
1923+
self.assertEqual(s.numIter, 1)
19211924

19221925

19231926
class KMeansTests(SparkSessionTestCase):

0 commit comments

Comments
 (0)