undid filter

Author: imatiach-msft

mllib/src/main/scala/org/apache/spark/mllib/evaluation/MulticlassMetrics.scala

@@ -54,17 +54,15 @@ class MulticlassMetrics @Since("3.0.0") (predAndLabelsWithOptWeight: RDD[_]) {
     .collectAsMap()
   private lazy val labelCount: Double = labelCountByClass.values.sum
   private lazy val tpByClass: Map[Double, Double] = predLabelsWeight
-    .filter(predLabelWeight => predLabelWeight._1 == predLabelWeight._2)
     .map {
-      case (_: Double, label: Double, weight: Double) =>
-        (label, weight)
+      case (prediction: Double, label: Double, weight: Double) =>
+        (label, if (label == prediction) weight else 0.0)
     }.reduceByKey(_ + _)
     .collectAsMap()
   private lazy val fpByClass: Map[Double, Double] = predLabelsWeight
-    .filter(predLabelWeight => predLabelWeight._1 != predLabelWeight._2)
     .map {
-      case (prediction: Double, _: Double, weight: Double) =>
-        (prediction, weight)
+      case (prediction: Double, label: Double, weight: Double) =>
+        (prediction, if (prediction != label) weight else 0.0)
     }.reduceByKey(_ + _)
     .collectAsMap()
   private lazy val confusions = predLabelsWeight