[ https://issues.apache.org/jira/browse/SPARK-21268?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Apache Spark reassigned SPARK-21268: ------------------------------------ Assignee: Apache Spark > Move center calculations to a distributed map in KMeans > ------------------------------------------------------- > > Key: SPARK-21268 > URL: https://issues.apache.org/jira/browse/SPARK-21268 > Project: Spark > Issue Type: Improvement > Components: MLlib > Affects Versions: 2.0.0, 2.0.1, 2.0.2, 2.1.0, 2.1.1 > Reporter: Guillaume Dardelet > Assignee: Apache Spark > Priority: Trivial > Labels: beginner, easyfix, newbie > Original Estimate: 1h > Remaining Estimate: 1h > > As I was monitoring the perfomance of my algorithm with SparkUI, I noticed > that their was a "collectAsMap" operation that was done hundreds of time at > every iteration of Kmeans: > https://github.com/apache/spark/blob/master/mllib/src/main/scala/org/apache/spark/mllib/clustering/KMeans.scala#L295 > It would work just as well by performing the following "foreach" on the RDD, > and would slightly improve perfomance. > Edit: > Per Sean Owen recommendations, scal() and VectorWithNorm creation should be > computed in a distributed map before the collectAsMap. -- This message was sent by Atlassian JIRA (v6.4.14#64029) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org