From 3a064ebd3d4e37aa33d9b43a0340502fe88578d8 Mon Sep 17 00:00:00 2001
From: bay9355 <mia.le@studium.uni-hamburg.de>
Date: Thu, 2 Mar 2023 15:56:12 +0100
Subject: [PATCH] =?UTF-8?q?Berechnung=20der=20betweenness=20beim=20erstell?=
 =?UTF-8?q?en=20des=20Graphen=20eingef=C3=BCgt?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cami_src/preprocess.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cami_src/preprocess.py b/cami_src/preprocess.py
index 5b54d82..8cd0a37 100644
--- a/cami_src/preprocess.py
+++ b/cami_src/preprocess.py
@@ -1,4 +1,5 @@
 import graph_tool
+import graph_tool.centrality
 
 def csv2graph(inputfile, 
               symbol_columns,
@@ -32,7 +33,7 @@ def csv2graph(inputfile,
                 unseen_vertices -= 1
             if unseen_vertices == 0:
                 break
-
+    g.vertex_properties["betweenness"], g.edge_properties["betweenness"] = graph_tool.centrality.betweenness(g)
     g.vertex_properties["cami_score"] = g.new_vertex_property("float", val=0.0)
     values = (20) * [-1]
     g.vertex_properties["predicted_by"] = g.new_vertex_property("vector<int16_t>", val=values)
-- 
GitLab