get raw vectors for further processing

apache · Nov 17, 2014 · 3782fa9 · 3782fa9
1 parent f1e7d1c
commit 3782fa9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/feature/Word2Vec.scala b/mllib/src/main/scala/org/apache/spark/mllib/feature/Word2Vec.scala
@@ -461,4 +461,13 @@ class Word2VecModel private[mllib] (
       .tail
       .toArray
   }
-}
+
+  /**
+   * Returns the strings with it´s raw vectors for further processing
+   * (e.g. clustering)
+   * @return map of string to vector
+   */
+  def getVectors(): Map[String, Array[Float]] = {
+    model
+  }
+}