diff --git a/netbone/structural/high_salience_skeleton.py b/netbone/structural/high_salience_skeleton.py index 80b2da884eb9ba5e9eb1eda1dedbd7a8f71469d1..ffec1ccaea0322768ea4628d21fd60f48691abd9 100644 --- a/netbone/structural/high_salience_skeleton.py +++ b/netbone/structural/high_salience_skeleton.py @@ -17,7 +17,7 @@ def high_salience_skeleton(data): values = {pair: 1 / wes[pair] for pair in wes} nx.set_edge_attributes(graph, values, name='distance') - nx.set_edge_attributes(graph, 0, name='score') + nx.set_edge_attributes(graph, 0, name='salience') for source in graph.nodes(): tree = nx.single_source_dijkstra(graph, source, cutoff=None, weight='distance')[1] @@ -29,9 +29,9 @@ def high_salience_skeleton(data): node_tree_scores[(path[i], path[i + 1])] = 1 for u, v in node_tree_scores: - graph[u][v]['score'] += 1 + graph[u][v]['salience'] += 1 - scores = nx.get_edge_attributes(graph, 'score') + scores = nx.get_edge_attributes(graph, 'salience') N = len(graph) score_values = dict() backbone_edges = dict()