commit 98d964ce18d42f7ba22331d092e9f943bd32c798
author Kenneth Chan <kenneth@prediction.io> Mon Mar 09 17:34:27 2015 -0700
committer Kenneth Chan <kenneth@prediction.io> Mon Mar 09 17:34:27 2015 -0700
tree 4a224bfc79be3263f89e99f51713c506d2584717
parent 67335ee8d0553d7ee298fc69002de4d3af1ecb23
parent 36202995b9843f2ee7822c58510d6aabf7822d26

Merge branch 'develop'

README.md

diff --git a/README.md b/README.md
index 08690bc..3b21e08 100644
--- a/README.md
+++ b/README.md
@@ -6,6 +6,10 @@
 
 ## Versions
 
+### v0.1.3
+
+- cache mllibRatings RDD in algorithm train() because it is used at multiple places (non-empty data check and ALS)
+
 ### v0.1.2
 
 - update for PredictionIO 0.9.0

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,