commit dc2fa910914a86c72e287d740379901ab3f3773e
author Kenneth Chan <kenneth@prediction.io> Mon Mar 09 17:30:33 2015 -0700
committer Kenneth Chan <kenneth@prediction.io> Mon Mar 09 17:30:33 2015 -0700
tree d102017393770f11491b6c1d5ec6607251e3b42e
parent ef0d50ba3f8492bf70139438596486d6c7873b2f

cache() mllibRatings output in algorithm train()

src/main/scala/ALSAlgorithm.scala

diff --git a/src/main/scala/ALSAlgorithm.scala b/src/main/scala/ALSAlgorithm.scala
index 358291e..7c0ddb9 100644
--- a/src/main/scala/ALSAlgorithm.scala
+++ b/src/main/scala/ALSAlgorithm.scala
@@ -90,6 +90,7 @@
         // MLlibRating requires integer index for user and item
         MLlibRating(u, i, v)
       }
+      .cache()
 
     // MLLib ALS cannot handle empty training data.
     require(!mllibRatings.take(1).isEmpty,