summaryrefslogtreecommitdiff
path: root/src/Chapter4/centrality/examples/BetweennessCentralityExample.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/Chapter4/centrality/examples/BetweennessCentralityExample.java')
-rw-r--r--src/Chapter4/centrality/examples/BetweennessCentralityExample.java31
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);
+
+ }
+}