checkpoint products 3,6,9 ...

apache · Jul 30, 2014 · ea25033 · ea25033
1 parent 154dccf
commit ea25033
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala b/mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala
@@ -255,7 +255,7 @@ class ALS private (
           rank, lambda, alpha, YtY)
         previousProducts.unpersist()
         logInfo("Re-computing U given I (Iteration %d/%d)".format(iter, iterations))
-        if (sc.checkpointDir.isDefined && (iter % 3 == 1)) {
+        if (sc.checkpointDir.isDefined && (iter % 3 == 0)) {
           products.checkpoint()
         }
         products.setName(s"products-$iter").persist()
@@ -271,7 +271,7 @@ class ALS private (
         logInfo("Re-computing I given U (Iteration %d/%d)".format(iter, iterations))
         products = updateFeatures(numProductBlocks, users, userOutLinks, productInLinks,
           rank, lambda, alpha, YtY = None)
-        if (sc.checkpointDir.isDefined && (iter % 3 == 1)) {
+        if (sc.checkpointDir.isDefined && (iter % 3 == 0)) {
           products.checkpoint()
         }
         products.setName(s"products-$iter")