diff options
author | Reza Zadeh <rizlar@gmail.com> | 2014-01-04 12:44:04 -0800 |
---|---|---|
committer | Reza Zadeh <rizlar@gmail.com> | 2014-01-04 12:44:04 -0800 |
commit | cdff9fc858b9b83eb1119ec2a6d1d3c9a66f47a9 (patch) | |
tree | d426962afa500f0f73771d49d5f9335f70ab9d90 /mllib/src/test | |
parent | e9bd6cb51dce9222a5a284cd171b299b0169852b (diff) | |
download | spark-cdff9fc858b9b83eb1119ec2a6d1d3c9a66f47a9.tar.gz spark-cdff9fc858b9b83eb1119ec2a6d1d3c9a66f47a9.tar.bz2 spark-cdff9fc858b9b83eb1119ec2a6d1d3c9a66f47a9.zip |
prettify
Diffstat (limited to 'mllib/src/test')
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/mllib/linalg/SVDSuite.scala | 34 |
1 files changed, 17 insertions, 17 deletions
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/linalg/SVDSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/linalg/SVDSuite.scala index a83d9d015b..4126e819e3 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/linalg/SVDSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/linalg/SVDSuite.scala @@ -47,25 +47,25 @@ class SVDSuite extends FunSuite with BeforeAndAfterAll { // Return jblas matrix from sparse matrix RDD def getDenseMatrix(matrix:RDD[MatrixEntry], m:Int, n:Int) : DoubleMatrix = { val ret = DoubleMatrix.zeros(m, n) - matrix.toArray.map(x => ret.put(x.i-1, x.j-1, x.mval)) + matrix.toArray.map(x => ret.put(x.i - 1, x.j - 1, x.mval)) ret } def assertMatrixEquals(a:DoubleMatrix, b:DoubleMatrix) { assert(a.rows == b.rows && a.columns == b.columns, "dimension mismatch") val diff = DoubleMatrix.zeros(a.rows, a.columns) - Array.tabulate(a.rows, a.columns){(i,j) => - diff.put(i,j, - Math.min(Math.abs(a.get(i,j)-b.get(i,j)), - Math.abs(a.get(i,j)+b.get(i,j)))) } + Array.tabulate(a.rows, a.columns){(i, j) => + diff.put(i, j, + Math.min(Math.abs(a.get(i, j) - b.get(i, j)), + Math.abs(a.get(i, j) + b.get(i, j)))) } assert(diff.norm1 < EPSILON, "matrix mismatch: " + diff.norm1) } test("full rank matrix svd") { val m = 10 val n = 3 - val data = sc.makeRDD(Array.tabulate(m,n){ (a,b)=> - MatrixEntry(a+1,b+1, (a+2).toDouble*(b+1)/(1+a+b)) }.flatten ) + val data = sc.makeRDD(Array.tabulate(m,n){ (a, b) => + MatrixEntry(a + 1, b + 1, (a + 2).toDouble * (b + 1) / (1 + a + b)) }.flatten ) val decomposed = SVD.sparseSVD(data, m, n, n) val u = decomposed.U @@ -75,9 +75,9 @@ class SVDSuite extends FunSuite with BeforeAndAfterAll { val densea = getDenseMatrix(data, m, n) val svd = Singular.sparseSVD(densea) - val retu = getDenseMatrix(u,m,n) - val rets = getDenseMatrix(s,n,n) - val retv = getDenseMatrix(v,n,n) + val retu = getDenseMatrix(u, m, n) + val rets = getDenseMatrix(s, n, n) + val retv = getDenseMatrix(v, n, n) // check individual decomposition assertMatrixEquals(retu, svd(0)) @@ -106,9 +106,9 @@ class SVDSuite extends FunSuite with BeforeAndAfterAll { val densea = getDenseMatrix(data, m, n) val svd = Singular.sparseSVD(densea) - val retu = getDenseMatrix(u,m,retrank) - val rets = getDenseMatrix(s,retrank,retrank) - val retv = getDenseMatrix(v,n,retrank) + val retu = getDenseMatrix(u, m, retrank) + val rets = getDenseMatrix(s, retrank, retrank) + val retv = getDenseMatrix(v, n, retrank) // check individual decomposition assertMatrixEquals(retu, svd(0).getColumn(0)) @@ -123,7 +123,7 @@ class SVDSuite extends FunSuite with BeforeAndAfterAll { val m = 10 val n = 3 val data = sc.makeRDD(Array.tabulate(m,n){ (a, b) => - MatrixEntry(a + 1, b + 1, (a + 2).toDouble*(b + 1)/(1 + a + b)) }.flatten ) + MatrixEntry(a + 1, b + 1, (a + 2).toDouble * (b + 1)/(1 + a + b)) }.flatten ) val k = 1 // only one svalue above this @@ -136,9 +136,9 @@ class SVDSuite extends FunSuite with BeforeAndAfterAll { val densea = getDenseMatrix(data, m, n) val svd = Singular.sparseSVD(densea) - val retu = getDenseMatrix(u,m,retrank) - val rets = getDenseMatrix(s,retrank,retrank) - val retv = getDenseMatrix(v,n,retrank) + val retu = getDenseMatrix(u, m, retrank) + val rets = getDenseMatrix(s, retrank, retrank) + val retv = getDenseMatrix(v, n, retrank) assert(retrank == 1, "rank returned not one") |