From 3782fa902b8700fd96619a36f5e23baf6b3a9a31 Mon Sep 17 00:00:00 2001
From: tkaessmann <tobias.kaessmann@s24.com>
Date: Mon, 17 Nov 2014 10:09:02 +0100
Subject: [PATCH] get raw vectors for further processing

---
 .../org/apache/spark/mllib/feature/Word2Vec.scala     | 11 ++++++++++-
 1 file changed, 10 insertions(+), 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
index f5f7ad613d4c4..fa6ef3644c2bc 100644
--- 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
+  }
+}
\ No newline at end of file