We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 967d041 commit f7a3ca2Copy full SHA for f7a3ca2
mllib/src/main/scala/org/apache/spark/mllib/rdd/VectorRDDFunctions.scala
@@ -59,16 +59,16 @@ private class Aggregator(
59
60
override def max(): Vector = {
61
nnz.activeIterator.foreach {
62
- case (id, 0.0) => currMax(id) = 0.0
63
- case _ =>
+ case (id, count) =>
+ if ((count == 0.0) || ((count < totalCnt) && (currMax(id) < 0.0))) currMax(id) = 0.0
64
}
65
Vectors.fromBreeze(currMax)
66
67
68
override def min(): Vector = {
69
70
- case (id, 0.0) => currMin(id) = 0.0
71
+ if ((count == 0.0) || ((count < totalCnt) && (currMin(id) > 0.0))) currMin(id) = 0.0
72
73
Vectors.fromBreeze(currMin)
74
0 commit comments