如何在R中展开使用igraph包制作的社区图

时间:2015-02-25 09:46:07

标签: r cluster-analysis igraph graph-visualization

尝试在推文数据中查找社区。不同单词之间的余弦相似性形成邻接矩阵。然后,我从该邻接矩阵中创建了图形。图形的可视化是这里的任务:

# Document Term Matrix
dtm = DocumentTermMatrix(tweets)

### adjust threshold here
dtms = removeSparseTerms(dtm, 0.998)
dim(dtms)

# cosine similarity matrix
t = as.matrix(dtms)

# comparing two word feature vectors
#cosine(t[,"yesterday"], t[,"yet"]) 

numWords = dim(t)[2]

# cosine measure between all column vectors of a matrix.
adjMat = cosine(t)

r = 3
for(i in 1:numWords)
{
  highElement  = sort(adjMat[i,], partial=numWords-r)[numWords-r]
  adjMat[i,][adjMat[i,] <  highElement] = 0
}

# build graph from the adjacency matrix
g = graph.adjacency(adjMat, weighted=TRUE, mode="undirected", diag=FALSE)
V(g)$name

# remove loop and multiple edges
g = simplify(g)
wt = walktrap.community(g, steps=5) # default steps=2
    table(membership(wt))

# set vertex color & size
nodecolor = rainbow(length(table(membership(wt))))[as.vector(membership(wt))]
nodesize = as.matrix(round((log2(10*membership(wt)))))
nodelayout = layout.fruchterman.reingold(g,niter=1000,area=vcount(g)^1.1,repulserad=vcount(g)^10.0, weights=NULL)

par(mai=c(0,0,1,0)) 
plot(g, 
     layout=nodelayout,
     vertex.size = nodesize,
     vertex.label=NA,
     vertex.color = nodecolor,
     edge.arrow.size=0.2,
     edge.color="grey",
     edge.width=1)

我只是希望在不同的群集/社区之间有更多的差距。

different communities are shown by different colors

1 个答案:

答案 0 :(得分:5)

据我所知,您不能仅使用igraph来布局相同社区的顶点彼此靠近。我已经在我的包NetPathMiner中实现了这个功能。看来,仅为可视化功能安装软件包有点困难。我将在这里写一个简单的版本并解释它的作用。

layout.by.attr <- function(graph, wc, cluster.strength=1,layout=layout.auto) {  
        g <- graph.edgelist(get.edgelist(graph)) # create a lightweight copy of graph w/o the attributes.
        E(g)$weight <- 1

        attr <- cbind(id=1:vcount(g), val=wc)
        g <- g + vertices(unique(attr[,2])) + igraph::edges(unlist(t(attr)), weight=cluster.strength)

        l <- layout(g, weights=E(g)$weight)[1:vcount(graph),]
        return(l)
}

基本上,该函数添加了一个额外的顶点,该顶点连接到属于同一社区的所有顶点。布局基于新图计算。由于每个社区现在都由一个共同的顶点连接,因此它们倾向于聚集在一起。

正如Gabor在评论中所说,增加边缘权重也会产生类似的效果。该函数利用此信息,通过增加cluster.strength,创建的顶点与其社区之间的边缘被赋予更高的权重。

如果仍然不够,则通过在相同社区的所有顶点之间添加边(形成一个集团)来扩展此原则(在更连通的图上计算布局)。根据我的经验,这有点矫枉过正。