0

私はsparkのmllibライブラリの中にあるword2vec関数を使用しています。私は、出力として取得しています単語ベクトルを印刷したい「getVectors」機能 私のコードは次のようになりますと:[String、Array [Float]]マップを印刷する方法は?

import org.apache.spark._ 

import org.apache.spark.rdd._ 
import org.apache.spark.SparkContext._ 
import org.apache.spark.mllib.feature.{Word2Vec, Word2VecModel} 

object word2vec { 
    def main(args: Array[String]) { 
    val conf = new SparkConf().setAppName("word2vec") 
    val sc = new SparkContext(conf) 
    val input = sc.textFile("file:///home/snap-01/balance.csv").map(line => line.split(",").toSeq) 
    val word2vec = new Word2Vec() 
    val model = word2vec.fit(input) 
    model.save(sc, "myModelPath") 
    val sameModel = Word2VecModel.load(sc, "myModelPath") 
    val vec = sameModel.getVectors 
    print(vec) 
    } 
} 

私は取得しています「地図(バランス - > [Fする@ 2932e15f)

答えて

2

これを試してみてください。また

vec.foreach { case (key, values) => println("key " + key + " - " + values.mkString("-") 
} 
+0

ありがとうございました:) – Aditi

1

println(vec.mapValues(_.toList)) 

しかし、そうするために必要なメモリに注目してください。

+0

ありがとうございました:) – Aditi

関連する問題