diff options
author | Neville Li <neville@spotify.com> | 2014-07-07 15:06:14 -0700 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2014-07-07 15:08:10 -0700 |
commit | 5044ba60a92495124b97ee97b87a45ce46d6073e (patch) | |
tree | d1e6acf4987d90d5de9c6a3db45b16a0f5dd2cd9 /examples | |
parent | b459aa77f63d0a469dc20e0ef555cf94382f41ca (diff) | |
download | spark-5044ba60a92495124b97ee97b87a45ce46d6073e.tar.gz spark-5044ba60a92495124b97ee97b87a45ce46d6073e.tar.bz2 spark-5044ba60a92495124b97ee97b87a45ce46d6073e.zip |
[SPARK-1977][MLLIB] register mutable BitSet in MovieLenseALS
Author: Neville Li <neville@spotify.com>
Closes #1319 from nevillelyh/gh/SPARK-1977 and squashes the following commits:
1f0a355 [Neville Li] [SPARK-1977][MLLIB] register mutable BitSet in MovieLenseALS
(cherry picked from commit f7ce1b3b48f0354434456241188c6a5d954852e2)
Signed-off-by: Xiangrui Meng <meng@databricks.com>
Diffstat (limited to 'examples')
-rw-r--r-- | examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala b/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala index 6eb41e7ba3..7a5a4cc2e7 100644 --- a/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala +++ b/examples/src/main/scala/org/apache/spark/examples/mllib/MovieLensALS.scala @@ -17,6 +17,8 @@ package org.apache.spark.examples.mllib +import scala.collection.mutable + import com.esotericsoftware.kryo.Kryo import org.apache.log4j.{Level, Logger} import scopt.OptionParser @@ -41,6 +43,7 @@ object MovieLensALS { class ALSRegistrator extends KryoRegistrator { override def registerClasses(kryo: Kryo) { kryo.register(classOf[Rating]) + kryo.register(classOf[mutable.BitSet]) } } |