[jira] [Created] (SPARK-7423) spark.ml Classifier predict should not convert vectors to dense format

["Joseph K. Bradley (JIRA)" <ji...@apache.org>]

Joseph K. Bradley created SPARK-7423:
----------------------------------------

             Summary: 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


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