diff options
Diffstat (limited to 'src/Chapter4/centrality/examples/BetweennessCentralityExample.java')
-rw-r--r-- | src/Chapter4/centrality/examples/BetweennessCentralityExample.java | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/src/Chapter4/centrality/examples/BetweennessCentralityExample.java b/src/Chapter4/centrality/examples/BetweennessCentralityExample.java new file mode 100644 index 0000000..ab9f7e6 --- /dev/null +++ b/src/Chapter4/centrality/examples/BetweennessCentralityExample.java @@ -0,0 +1,31 @@ +package centrality.examples; + +import Chapter4.util.TweetFileToGraph; +import java.io.File; +import GraphElements.RetweetEdge; +import GraphElements.UserNode; +import edu.uci.ics.jung.algorithms.importance.BetweennessCentrality; +import edu.uci.ics.jung.graph.DirectedGraph; + +public class BetweennessCentralityExample { + public static void main(String[] args){ + + File tweetFile; + + if(args.length > 0){ + tweetFile = new File(args[0]); + } + else{ + tweetFile = new File("synthetic_retweet_network.json"); + } + + DirectedGraph<UserNode, RetweetEdge> retweetGraph = TweetFileToGraph.getRetweetNetwork(tweetFile); + + //calculate the betweenness centrality + BetweennessCentrality<UserNode, RetweetEdge> betweenness = new BetweennessCentrality<UserNode, RetweetEdge>(retweetGraph); + + betweenness.evaluate(); + betweenness.printRankings(true, true); + + } +} |