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 fa6ef3644c2bc..34d8245d3fbcf 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 @@ -463,11 +463,11 @@ class Word2VecModel private[mllib] ( } /** - * Returns the strings with it´s raw vectors for further processing + * 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]] = { + def getVectors: Map[String, Array[Float]] = { model } -} \ No newline at end of file +}