[
https://issues.apache.org/jira/browse/SPARK-22737?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=16283048#comment-16283048
]
Apache Spark commented on SPARK-22737:
--------------------------------------
User 'zhengruifeng' has created a pull request for this issue:
https://github.com/apache/spark/pull/19927
> Simplity OneVsRest transform
> ----------------------------
>
> Key: SPARK-22737
> URL: https://issues.apache.org/jira/browse/SPARK-22737
> Project: Spark
> Issue Type: Improvement
> Components: ML
> Affects Versions: 2.3.0
> Reporter: zhengruifeng
>
> Current impl of OneVsRest#transform is over-complicated. It sequentially
> updates an acumulated column.
> By using a direct UDF of prediction, we obtain a speedup of at least 2x.
> On some extreme case with 20 classes, it obtain about 14x speedup.
> The test code and performance comparsion details are in the corresponding PR.
--
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
