From 7ca733beeb18808e145dc2786f9c2c6c1ec40031 Mon Sep 17 00:00:00 2001 From: Huaxin Gao Date: Wed, 27 Jun 2018 18:08:49 -0700 Subject: [PATCH] overload the set/getDistanceMeasure methods in BisectingKMeans --- python/pyspark/ml/clustering.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/python/pyspark/ml/clustering.py b/python/pyspark/ml/clustering.py index d2dffca4480dd..6d77baf7349e4 100644 --- a/python/pyspark/ml/clustering.py +++ b/python/pyspark/ml/clustering.py @@ -658,6 +658,20 @@ def getMinDivisibleClusterSize(self): """ return self.getOrDefault(self.minDivisibleClusterSize) + @since("2.4.0") + def setDistanceMeasure(self, value): + """ + Sets the value of :py:attr:`distanceMeasure`. + """ + return self._set(distanceMeasure=value) + + @since("2.4.0") + def getDistanceMeasure(self): + """ + Gets the value of `distanceMeasure` or its default value. + """ + return self.getOrDefault(self.distanceMeasure) + def _create_model(self, java_model): return BisectingKMeansModel(java_model)