diff options
author | Reza Zadeh <rizlar@gmail.com> | 2014-01-04 14:28:07 -0800 |
---|---|---|
committer | Reza Zadeh <rizlar@gmail.com> | 2014-01-04 14:28:07 -0800 |
commit | 06c0f7628a213a08ef5adeab903160b806680acf (patch) | |
tree | 3343a151864808dabbc15ba481f125e6723ab55e /examples/src | |
parent | cdff9fc858b9b83eb1119ec2a6d1d3c9a66f47a9 (diff) | |
download | spark-06c0f7628a213a08ef5adeab903160b806680acf.tar.gz spark-06c0f7628a213a08ef5adeab903160b806680acf.tar.bz2 spark-06c0f7628a213a08ef5adeab903160b806680acf.zip |
use SparseMatrix everywhere
Diffstat (limited to 'examples/src')
-rw-r--r-- | examples/src/main/scala/org/apache/spark/examples/SparkSVD.scala | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/examples/src/main/scala/org/apache/spark/examples/SparkSVD.scala b/examples/src/main/scala/org/apache/spark/examples/SparkSVD.scala index 5590ee728a..4b9e674c68 100644 --- a/examples/src/main/scala/org/apache/spark/examples/SparkSVD.scala +++ b/examples/src/main/scala/org/apache/spark/examples/SparkSVD.scala @@ -20,6 +20,7 @@ package org.apache.spark.examples import org.apache.spark.SparkContext import org.apache.spark.mllib.linalg.SVD import org.apache.spark.mllib.linalg.MatrixEntry +import org.apache.spark.mllib.linalg.SparseMatrix /** * Compute SVD of an example matrix @@ -48,10 +49,10 @@ object SparkSVD { val n = 4 // recover largest singular vector - val decomposed = SVD.sparseSVD(data, m, n, 1) - val u = decomposed.U - val s = decomposed.S - val v = decomposed.V + val decomposed = SVD.sparseSVD(SparseMatrix(data, m, n), 1) + val u = decomposed.U.data + val s = decomposed.S.data + val v = decomposed.V.data println("singular values = " + s.toArray.mkString) } |