Added scala function to stringify userFeatures for access in python.

apache · Oct 2, 2014 · e1fbe5e · e1fbe5e
1 parent d3a3840
commit e1fbe5e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/recommendation/MatrixFactorizationModel.scala b/mllib/src/main/scala/org/apache/spark/mllib/recommendation/MatrixFactorizationModel.scala
@@ -106,4 +106,8 @@ class MatrixFactorizationModel private[mllib] (
     }
     scored.top(num)(Ordering.by(_._2))
   }
+
+  def userFeaturesString(userFeatures: RDD[(Int, Array[Double])]): RDD[String]  = {
+    userFeatures.map(element => element._1+","+element._2.mkString(","))
+  }
 }