minor fixes

Author: Ishiihara

mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala

@@ -298,12 +298,12 @@ class PythonMLLibAPI extends Serializable {
    * @return A handle to java Word2VecModelWrapper instance at python side
    */
   def trainWord2Vec(
-    dataJRDD: JavaRDD[java.util.ArrayList[String]],
-    vectorSize: Int,
-    learningRate: Double,
-    numPartitions: Int,
-    numIterations: Int,
-    seed: Long): Word2VecModelWrapper = {
+      dataJRDD: JavaRDD[java.util.ArrayList[String]],
+      vectorSize: Int,
+      learningRate: Double,
+      numPartitions: Int,
+      numIterations: Int,
+      seed: Long): Word2VecModelWrapper = {
     val data = dataJRDD.rdd.cache()
     val word2vec = new Word2Vec()
         .setVectorSize(vectorSize)

mllib/src/main/scala/org/apache/spark/mllib/feature/Word2Vec.scala

@@ -437,7 +437,7 @@ class Word2VecModel private[mllib] (
    * Find synonyms of a word
    * @param word a word
    * @param num number of synonyms to find  
-   * @return array of (word, similarity)
+   * @return array of (word, cosineSimilarity)
    */
   def findSynonyms(word: String, num: Int): Array[(String, Double)] = {
     val vector = transform(word)