Vincent created SPARK-22096:
-------------------------------
Summary: 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
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 ~20% performance gain with these changes.
--
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
