[
https://issues.apache.org/jira/browse/SPARK-7423?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Xiangrui Meng updated SPARK-7423:
---------------------------------
Assignee: George Dittmar
> 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
> Assignee: George Dittmar
> Priority: Minor
> Labels: starter
> Fix For: 1.5.0
>
>
> 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: [email protected]
For additional commands, e-mail: [email protected]
