[ https://issues.apache.org/jira/browse/SPARK-8587?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Sam Stoelinga updated SPARK-8587: --------------------------------- Description: Looking at PySpark the implementation of KMeansModel.predict https://github.com/apache/spark/blob/master/python/pyspark/mllib/clustering.py#L102 : Currently: it calculates the cost of the closest cluster and returns the index only. My expectation: Easy way to let the same function or a new function to return the cost with the index. was: Looking at PySpark the implementation of KMeansModel.predict https://github.com/apache/spark/blob/master/python/pyspark/mllib/clustering.py#L102: Currently: it calculates the cost of the closest cluster and returns the index only. My expectation: Easy way to let the same function or a new function to return the cost with the index. > Return cost and cluster index KMeansModel.predict > ------------------------------------------------- > > Key: SPARK-8587 > URL: https://issues.apache.org/jira/browse/SPARK-8587 > Project: Spark > Issue Type: Improvement > Reporter: Sam Stoelinga > Priority: Minor > > Looking at PySpark the implementation of KMeansModel.predict > https://github.com/apache/spark/blob/master/python/pyspark/mllib/clustering.py#L102 > : > Currently: > it calculates the cost of the closest cluster and returns the index only. > My expectation: > Easy way to let the same function or a new function to return the cost with > the index. -- This message was sent by Atlassian JIRA (v6.3.4#6332) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org