word2vec的spark实现_spark之word2vec使用(python)

from pyspark import SparkConf, SparkContext,SQLContext

from pyspark.sql import SparkSession

from pyspark.ml.feature import Word2Vec,CountVectorizer

conf = SparkConf().setAppName("yjs_rec")

sc = SparkContext(conf=conf)

sqlContext=SQLContext(sc)

'''

documentDF = sqlContext.createDataFrame([

("Hi I heard about Spark".split(" "),),

("I wish Java could use case classes".split(" "),),

("Logistic regression models are neat".split(" "),)

], ["text"])

'''

spark_df = sqlContext.createDataFrame(documentDF)

#word2vec

word2Vec = Word2Vec(vectorSize=100, minCount=0, inputCol="words", outputCol="result")

model = word2Vec.fit(spark_df)

result = model.transform(spark_df)

result.select("result").show()

分享到:

18e900b8666ce6f233d25ec02f95ee59.png

72dd548719f0ace4d5f9bca64e1d7715.png

2018-08-15 09:57

浏览 1746

评论

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值