You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@spark.apache.org by "George Dittmar (JIRA)" <ji...@apache.org> on 2015/05/26 05:25:17 UTC
[jira] [Comment Edited] (SPARK-7423) spark.ml Classifier predict
should not convert vectors to dense format
[ https://issues.apache.org/jira/browse/SPARK-7423?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14558651#comment-14558651 ]
George Dittmar edited comment on SPARK-7423 at 5/26/15 3:25 AM:
----------------------------------------------------------------
Still waiting on PR for spark-7422 to be accepted so this change can be done so its still in progress.
was (Author: georgedittmar):
Still waiting on PR for spark-7422 to be submitted so this change can be done so its still in progress.
> spark.ml Classifier predict should not convert vectors to dense format
> ----------------------------------------------------------------------
>
> Key: SPARK-7423
> URL: https://issues.apache.org/jira/browse/SPARK-7423
> Project: Spark
> Issue Type: Improvement
> Components: ML
> Reporter: Joseph K. Bradley
> Priority: Minor
> Labels: starter
>
> spark.ml.classification.ClassificationModel and ProbabilisticClassificationModel both use DenseVector.argmax to implement prediction (computing the prediction from the rawPrediction or probability Vectors). It would be best to implement argmax for Vector and SparseVector and use Vector.argmax, rather than converting Vectors to dense format.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org