使用Spark Streaming进行情感分析

来源:InfoQ 作者:侠天
  

 

这里将使用Twitter流式数据,它符合所有所需:持续而且无止境的数据源。

Spark Streaming

Spark Streaming在电子书《手把手教你学习Spark》第六章有详细介绍,这里略过Streaming API的详细介绍,直接进行程序开发 。

程序开发设置部分

程序开发起始部分需要做好准备工作。

val config = new SparkConf().setAppName("twitter-stream-sentiment")
val sc = new SparkContext(config)
sc.setLogLevel("WARN")

val ssc = new StreamingContext(sc, Seconds(5))

System.setProperty("twitter4j.oauth.consumerKey", "consumerKey")
System.setProperty("twitter4j.oauth.consumerSecret", "consumerSecret")
System.setProperty("twitter4j.oauth.accessToken", accessToken)
System.setProperty("twitter4j.oauth.accessTokenSecret", "accessTokenSecret")

val stream = TwitterUtils.createStream(ssc, None)

这里创建一个Spark Context sc,设置日志级别为WARN来消除Spark生成的日志。使用sc创建Streaming Contextssc,然后设置 Twitter证书来获得 Twitter网站数据。

Twitter上现在的趋势是什么?

很容易的能够找到任意给定时刻的Twitter趋势,仅仅需要计算数据流每个标签的数目。让我们看下Spark如何实现这个操作的。

val tags = stream.flatMap { status =>
   status.getHashtagEntities.map(_.getText)
}
tags.countByValue()
   .foreachRDD { rdd =>
       val now = org.joda.time.DateTime.now()
       rdd
         .sortBy(_._2)
         .map(x => (x, now))
         .saveAsTextFile(s"~/twitter/$now")
     }

首先从Tweets获取标记,并计算标记的数量,按数量排序,然后持久化结果。我们基于前面的结果建立一个监控面板来跟踪趋势标签。作者的同事就可以创建一个广告标记(campaigns),并吸引更多的用户。

分析Tweets

现在我们想增加一个功能来获得用户主要感兴趣的主题集。为了这个目的我们想对Tweets的大数据和食物两个不相关的主题进行情感分析。

有几种API可以在Tweets上做情感分析,但是作者选择斯坦福自然语言处理组开发的库来抽取相关情感。
build.sbt文件中增加相对应的依赖。

libraryDependencies += "edu.stanford.nlp" % "stanford-corenlp" % "3.5.1"
libraryDependencies += "edu.stanford.nlp" % "stanford-corenlp" % "3.5.1" classifier "models"

现在,我们通过Streaming过滤一定的哈希标签,只选择感兴趣的Tweets,如下所示:

val tweets = stream.filter {t =>
     val tags = t.getText.split(" ").filter(_.startsWith("#")).map(_.toLowerCase)
     tags.contains("#bigdata") && tags.contains("#food")
   }

得到Tweets上所有标签,然后标记出#bigdata和 #food两个标签。
接下来定一个函数从Tweets抽取相关的情感:

def detectSentiment(message: String): SENTIMENT_TYPE

然后对detectSentiment进行测试以确保其可以工作:

it("should detect not understood sentiment") {
     detectSentiment("") should equal (NOT_UNDERSTOOD)
}

it("should detect a negative sentiment") {
     detectSentiment("I am feeling very sad and frustrated.") should equal (NEGATIVE)
}

it("should detect a neutral sentiment") {
     detectSentiment("I'm watching a movie") should equal (NEUTRAL)
}

it("should detect a positive sentiment") {
     detectSentiment("It was a nice experience.") should equal (POSITIVE)
}

it("should detect a very positive sentiment") {
     detectSentiment("It was a very nice experience.") should equal (VERY_POSITIVE)
}

完整列子如下:

val data = tweets.map { status =>
   val sentiment = SentimentAnalysisUtils.detectSentiment(status.getText)
   val tags = status.getHashtagEntities.map(_.getText.toLowerCase)

   (status.getText, sentiment.toString, tags)
}

data中包含相关的情感。

和SQL协同进行分析

现在作者想把情感分析的数据存储在外部数据库,为了后续可以使用SQL查询。
具体操作如下:

val sqlContext = new SQLContext(sc)

import sqlContext.implicits._

data.foreachRDD { rdd =>
   rdd.toDF().registerTempTable("sentiments")
}

将Dstream转换成DataFrame,然后注册成一个临时表,其他喜欢使用SQL的同事就可以使用不同的数据源啦。

sentiment表可以被任意查询,也可以使用Spark SQL和其他数据源(比如,Cassandra数据等)进行交叉查询。
查询DataFrame的列子:

sqlContext.sql("select * from sentiments").show()

窗口操作

Spark Streaming的窗口操作可以进行回溯数据,这在其他流式引擎中并没有。
为了使用窗口函数,你需要checkpoint流数据,具体详情见
http://spark.apache.org/docs/latest/streaming-programming-guide.html#checkpointing
简单的一个窗口操作:

tags
   .window(Minutes(1))
   . (...)

结论

此列子虽然简单,但是其可以使用Spark解决实际问题。我们可以计算Twitter上主题趋势。


时间:2016-05-25 08:31 来源:InfoQ 作者:侠天 原文链接

好文,顶一下
(0)
0%
文章真差,踩一下
(0)
0%
------分隔线----------------------------


把开源带在你的身边-精美linux小纪念品
无觅相关文章插件,快速提升流量