add unit test for dostack ALS

apache · Jun 29, 2016 · 56194eb · 56194eb
1 parent 3607bdc
commit 56194eb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/recommendation/ALSSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/recommendation/ALSSuite.scala
@@ -389,6 +389,11 @@ class ALSSuite
       targetRMSE = 0.3)
   }
 
+  test("rank-129 matrix with stacking factors in matrices") {
+    val (training, test) = genExplicitTestData(numUsers = 200, numItems = 20, rank = 1)
+    testALS(training, test, maxIter = 1, rank = 129, regParam = 0.01, targetRMSE = 0.02)
+  }
+
   test("using generic ID types") {
     val (ratings, _) = genImplicitTestData(numUsers = 20, numItems = 40, rank = 2, noiseStd = 0.01)