Description
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.
Attachments
Issue Links
- requires
-
SPARK-7422 Add argmax to Vector, SparseVector
-
- Closed
-
- links to