diff --git a/cami_src/preprocess.py b/cami_src/preprocess.py index 5b54d82b7946eee99224983b204e361fffd995b8..8cd0a37016386efb9faaa0cb381e34e7db34db4c 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)