filter zeros in sparse

andrewor14 · Mar 20, 2014 · 0dc7980 · 0dc7980
1 parent 6115610
commit 0dc7980
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/util/LAUtils.scala b/mllib/src/main/scala/org/apache/spark/mllib/util/LAUtils.scala
@@ -58,6 +58,7 @@ object LAUtils {
     val n = a.n
     val data = a.rows.flatMap{
       mrow => Array.tabulate(n)(j => MatrixEntry(mrow.i, j, mrow.data(j)))
+                   .filter(x => x.mval != 0)
     }
     SparseMatrix(data, m, n)
   }