原文链接：关联分析(3):Apriori R语言实现

微信公众号：机器学习养成记    搜索添加微信公众号：chenchenwings

---

在关联分析(1):概念及应用中我们介绍了关联分析的基本概念及应用场景，关联分析(2):Apriori产生频繁项集我们介绍了Apriori算法的原理，本篇文章我们将进行关联分析的R语言演示。

R语言实现

我们对UCI机器学习库上下载的美国众议院议员投票记录数据，进行关联分析。在R中，可以直接调用arules包中的apriori()函数训练模型。

导入包与数据。arulesViz包可以将关联规则进行可视化输出，“header=F”设置读入数据时，首行不作为列名。

library(arules)

library(arulesViz)

data<-read.csv("D\\voting-records.csv", header=F)

训练模型和性能评估。在训练关联分析模型时，需要设置支持度和置信度参数，这里，我们设定支持度为30%，置信度为90%。“minlen = 2”表示选取至少包含两个项的规则，以避免得到由于某项出现过于频繁而创建的无用规则。

data_rules <- apriori(data = data, parameter = list(support = 0.3, confidence = 0.9, minlen = 2))

summary(data_rules)

性能评估结果如下：

其中rule length distribution 表示规则长度分布，如前件+后件共包含两项的规则有39条。

查看规则。inspect()可以查看具体规则。首先得出根据支持度排序，位于前十位的规则。通过subset()函数，我们可以找到与某一项相关的规则。

%inspect(sort(data_rules, by="support")[1:10])

%与V15=crime:n相关的规则

crime_n<-subset(data_rules, items %in% "V15=crime:n")

inspect(sort(crime_n, by="support")[1:10])

%in%表示精确匹配。最终得到结果为：

lift为提升度，表示前件存在后件发生的概率，是前件不存在但后件发生的倍数。比如第一条规则，{V15=crime:n} => {V1=democrat} 的提升度为1.6，表示{V15=crime:n}发生后 {V1=democrat}也发生，是{V15=crime:n} 不发生但{V1=democrat}发生的1.6倍。

结果可视化。可以看出规则的三个参数之间关系。

shading = "lift" 表示在散点图上颜色深浅的度量是lift。

jitter=2表示增加抖动值。measure="confidence"表示圆圈大小受置信度影响。

plot(data_rules, control=list(jitter=2), shading = "lift")

plot(sort(data_rules, by="support")[1:10], method="grouped") 

plot(sort(crime_n, by="support")[1:10], measure="confidence", method="graph", control=list(type="items"), shading = "lift")

---

推荐文章

· Bagging算法(R语言)

·静态爬虫与地址经纬度转换(python)

· 特征工程(一):前向逐步回归(R语言)

·聚类(三):KNN算法(R语言)

· 小案例(六)：预测小偷行为（python）

·ggplot2:正负区分条形图及美化

扫描二维码，关注我们。

如需转载，请在开篇显著位置注明作者和出处，并在文末放置机器学习养成记二维码和添加原文链接。

快来关注我们吧！