[
https://issues.apache.org/jira/browse/SPARK-22096?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Hyukjin Kwon updated SPARK-22096:
---------------------------------
Labels: bulk-closed (was: )
> use aggregateByKeyLocally to save one stage in calculating ItemFrequency in
> NaiveBayes
> --------------------------------------------------------------------------------------
>
> Key: SPARK-22096
> URL: https://issues.apache.org/jira/browse/SPARK-22096
> Project: Spark
> Issue Type: Improvement
> Components: ML
> Affects Versions: 2.2.0
> Reporter: Vincent
> Priority: Minor
> Labels: bulk-closed
> Attachments: performance data for NB.png
>
>
> NaiveBayes currently takes aggreateByKey followed by a collect to calculate
> frequency for each feature/label. We can implement a new function
> 'aggregateByKeyLocally' in RDD that merges locally on each mapper before
> sending results to a reducer to save one stage.
> We tested on NaiveBayes and see ~16% performance gain with these changes.
> [^performance data for NB.png]
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
