[ 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