From 1efcfee066f62f625cbeb280d5f4a18dbf67ce84 Mon Sep 17 00:00:00 2001 From: Yassin <aliyassin4@hotmail.com> Date: Wed, 10 May 2023 02:19:34 +0200 Subject: [PATCH] Refactorings Code and Adding Node Filter --- netbone/filters.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/netbone/filters.py b/netbone/filters.py index 2dacd36..a728f80 100644 --- a/netbone/filters.py +++ b/netbone/filters.py @@ -60,16 +60,9 @@ 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) - - print(list(data[:value].index)) - b.remove_nodes_from(list(data[:value].index)) - return b + return b.subgraph(list(data[:value].index)) print("The accepted filters for " + backbone.method_name + " are: " + ', '.join( [fun.__name__ for fun in backbone.compatible_filters()])) -- GitLab