From ecd1c4116532c68b2c60f4bf8b46afdd2d2b4a2c Mon Sep 17 00:00:00 2001 From: Yassin <aliyassin4@hotmail.com> Date: Wed, 10 May 2023 02:14:29 +0200 Subject: [PATCH] Refactorings Code and Adding Node Filter --- netbone/filters.py | 4 ++++ netbone/structural/modulairy_backbone.py | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/netbone/filters.py b/netbone/filters.py index 4ce8fef..7d31580 100644 --- a/netbone/filters.py +++ b/netbone/filters.py @@ -60,8 +60,12 @@ def fraction_filter(backbone, value, narrate=True, secondary_property='weight', value = math.ceil(value * len(data)) return nx.from_pandas_edgelist(data[:value], edge_attr=edge_properties(data)) else: + print('hereeeeeeee') + b = backbone.graph.copy() value = math.ceil(value * len(backbone.graph)) + + print(value) b.remove_nodes_from(list(data[:value].index)) return b diff --git a/netbone/structural/modulairy_backbone.py b/netbone/structural/modulairy_backbone.py index c73de25..79e62bc 100644 --- a/netbone/structural/modulairy_backbone.py +++ b/netbone/structural/modulairy_backbone.py @@ -67,7 +67,7 @@ def modularity_backbone(data): vitalities = (modularity_value - q1s).tolist() - nx.set_node_attributes(g, dict(zip(list(g.nodes()), np.absolute(vitalities))), name='score') + nx.set_node_attributes(g, dict(zip(list(g.nodes()), np.absolute(vitalities))), name='vitality') - return Backbone(g, method_name="Modularity Filter", property_name='score', ascending=False, + return Backbone(g, method_name="Modularity Filter", property_name='vitality', ascending=False, compatible_filters=[threshold_filter, fraction_filter], filter_on='Nodes') -- GitLab