GitHub user Ishiihara opened a pull request:
https://github.com/apache/spark/pull/1932
[SPARK-2907][MLlib] Word2Vec performance improve
@mengxr Please review the code. Adding weights in reduceByKey soon.
Only output model entry for words appeared in the partition before merging
and use reduceByKey to combine model. In general, this implementation is 30s or
so faster than implementation using big array.
You can merge this pull request into a Git repository by running:
$ git pull https://github.com/Ishiihara/spark Word2Vec-improve2
Alternatively you can review and apply these changes as the patch at:
https://github.com/apache/spark/pull/1932.patch
To close this pull request, make a commit to your master/trunk branch
with (at least) the following in the commit message:
This closes #1932
----
commit aa2ab36c6a9fa22e759ccb99352394dd6d6317e0
Author: Liquan Pei <[email protected]>
Date: 2014-08-13T12:45:17Z
use reduceByKey to combine models
commit 9075e1cba5ae64add2986514be99dc51083ff177
Author: Liquan Pei <[email protected]>
Date: 2014-08-13T23:17:58Z
combine syn0Global and syn1Global to synGlobal
----
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]