[ https://issues.apache.org/jira/browse/SPARK-21268?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16070300#comment-16070300 ]
Apache Spark commented on SPARK-21268: -------------------------------------- User 'dardelet' has created a pull request for this issue: https://github.com/apache/spark/pull/18491 > 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 > 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