diff options
author | Jianping J Wang <jianping.j.wang@gmail.com> | 2013-12-30 23:42:55 +0800 |
---|---|---|
committer | Jianping J Wang <jianping.j.wang@gmail.com> | 2013-12-30 23:42:55 +0800 |
commit | 600421d8bc258508d0e298131384d0abed4b87b7 (patch) | |
tree | 77526668abee38d6bfa9e583f3b52879b67fd0b0 /graph/src | |
parent | 29fe6bdaa29193c9dbf3a8fbd05094f3d812d4e5 (diff) | |
download | spark-600421d8bc258508d0e298131384d0abed4b87b7.tar.gz spark-600421d8bc258508d0e298131384d0abed4b87b7.tar.bz2 spark-600421d8bc258508d0e298131384d0abed4b87b7.zip |
update
Diffstat (limited to 'graph/src')
-rw-r--r-- | graph/src/test/scala/org/apache/spark/graph/algorithms/SvdppSuite.scala | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/graph/src/test/scala/org/apache/spark/graph/algorithms/SvdppSuite.scala b/graph/src/test/scala/org/apache/spark/graph/algorithms/SvdppSuite.scala index 4ea675c2dc..46dee4e9a0 100644 --- a/graph/src/test/scala/org/apache/spark/graph/algorithms/SvdppSuite.scala +++ b/graph/src/test/scala/org/apache/spark/graph/algorithms/SvdppSuite.scala @@ -13,12 +13,13 @@ class SvdppSuite extends FunSuite with LocalSparkContext { test("Test SVD++ with mean square error on training set") { withSpark { sc => - val SvdppErr = 0.01 + val SvdppErr = 0.2 val edges = sc.textFile("mllib/data/als/test.data").map { line => val fields = line.split(",") Edge(fields(0).toLong * 2, fields(1).toLong * 2 + 1, fields(2).toDouble) } - val graph = Svdpp.run(edges) + val conf = new SvdppConf(10, 2, 0.0, 5.0, 0.007, 0.007, 0.005, 0.015) + val graph = Svdpp.run(edges, conf) val err = graph.vertices.collect.map{ case (vid, vd) => if (vid % 2 == 1) { vd.norm } else { 0.0 } }.reduce(_ + _) / graph.triplets.collect.size |