From 8f374163426cad4b5a58d9af43e7940f77536494 Mon Sep 17 00:00:00 2001
From: mahoffman <mahoffman@protonmail.com>
Date: Sun, 8 Oct 2017 11:07:38 -0400
Subject: [PATCH] visualize_fix

- Carried pruned through the remainder of the function.
- Changed communities$memberships to $membership (memberships is a matrix).
- Reordered plotting so that edges are plotted before nodes.
---
 R/visualize.R | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/R/visualize.R b/R/visualize.R
index 9409cb8..c33c1da 100644
--- a/R/visualize.R
+++ b/R/visualize.R
@@ -18,17 +18,17 @@ visualize<-function(text_network, prune_cut, label_degree_cut=0, betweenness=FAL
   isolates <- V(pruned)[degree(pruned)==0]
   pruned <- delete.vertices(pruned, isolates)
   #calculate modularity for coloring
-  communities<-cluster_louvain(text_network)
-  V(pruned)$modularity<-communities$memberships
+  communities <- cluster_louvain(pruned)
+  V(pruned)$modularity<-communities$membership
   #calculate betweenness for sizing nodes
   size=2
   if(betweenness){
-    size<-betweenness(text_network)
+    size<-betweenness(pruned)
   }
   #make visualization
   ggraph(pruned, layout = "fr") +
-    geom_node_point(color = V(pruned)$modularity, size = size) +
     geom_edge_link(aes(edge_alpha = weight), show.legend = FALSE)+
+    geom_node_point(color = V(pruned)$modularity, size = size) +
     geom_node_text(aes(label = name, filter=degree>label_degree_cut), repel = TRUE, size=2) +
     theme_void()
 }