diff options
-rw-r--r-- | mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java | 16 | ||||
-rw-r--r-- | mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala | 8 |
2 files changed, 12 insertions, 12 deletions
diff --git a/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java b/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java index c8e59f7399..eafee060cd 100644 --- a/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java +++ b/mllib/src/test/java/org/apache/spark/mllib/recommendation/JavaALSSuite.java @@ -102,8 +102,8 @@ public class JavaALSSuite implements Serializable { public void runALSUsingStaticMethods() { int features = 1; int iterations = 15; - int users = 10; - int products = 10; + int users = 50; + int products = 100; scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList( users, products, features, 0.7, false); @@ -116,8 +116,8 @@ public class JavaALSSuite implements Serializable { public void runALSUsingConstructor() { int features = 2; int iterations = 15; - int users = 20; - int products = 30; + int users = 100; + int products = 200; scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList( users, products, features, 0.7, false); @@ -133,8 +133,8 @@ public class JavaALSSuite implements Serializable { public void runImplicitALSUsingStaticMethods() { int features = 1; int iterations = 15; - int users = 40; - int products = 80; + int users = 80; + int products = 160; scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList( users, products, features, 0.7, true); @@ -147,8 +147,8 @@ public class JavaALSSuite implements Serializable { public void runImplicitALSUsingConstructor() { int features = 2; int iterations = 15; - int users = 50; - int products = 100; + int users = 100; + int products = 200; scala.Tuple3<List<Rating>, DoubleMatrix, DoubleMatrix> testData = ALSSuite.generateRatingsAsJavaList( users, products, features, 0.7, true); diff --git a/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala index 1ab181d35a..fafc5ec5f2 100644 --- a/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala +++ b/mllib/src/test/scala/org/apache/spark/mllib/recommendation/ALSSuite.scala @@ -87,19 +87,19 @@ class ALSSuite extends FunSuite with BeforeAndAfterAll { } test("rank-1 matrices") { - testALS(10, 20, 1, 15, 0.7, 0.3) + testALS(50, 100, 1, 15, 0.7, 0.3) } test("rank-2 matrices") { - testALS(20, 30, 2, 15, 0.7, 0.3) + testALS(100, 200, 2, 15, 0.7, 0.3) } test("rank-1 matrices implicit") { - testALS(40, 80, 1, 15, 0.7, 0.4, true) + testALS(80, 160, 1, 15, 0.7, 0.4, true) } test("rank-2 matrices implicit") { - testALS(50, 100, 2, 15, 0.7, 0.4, true) + testALS(100, 200, 2, 15, 0.7, 0.4, true) } /** |