Updating HSS

netbone/structural/high_salience_skeleton.py

@@ -36,6 +36,22 @@ def high_salience_skeleton(data):
 
     scores= nx.get_edge_attributes(graph, 'score')
     N = len(graph)
-    values = {pair:scores[pair]/N for pair in scores}
-    nx.set_edge_attributes(graph, values, name='score')
+    score_values = dict()
+    backbone_edges = dict()
+    for pair in scores:
+        score_values[pair] = scores[pair]/N
+        if scores[pair]/N > 0.8:
+            backbone_edges[pair] = True
+        else:
+            backbone_edges[pair] = False
+
+            # score_values = {pair:scores[pair]/N for pair in scores}
+    nx.set_edge_attributes(graph, score_values, name='score')
+    nx.set_edge_attributes(graph, backbone_edges, name='high_salience_skeleton')
+
+    # for u,v in graph.edges():
+    #     if graph[u][v]['score']>=0.8:
+    #         graph[u][v]['high_salience_skeleton'] = True
+    #     else:
+    #         graph[u][v]['high_salience_skeleton'] = False
     return Backbone(graph, name="High Salience Skeleton Filter", column="high_salience_skeleton", ascending=False, filters=[boolean_filter, threshold_filter, fraction_filter])