From 9a09b49d867a92d9b105bb1e2352d4d2af3fbcc0 Mon Sep 17 00:00:00 2001 From: Johannes Huegle Date: Thu, 9 Feb 2023 10:49:02 +0100 Subject: [PATCH] Ensure reliable edge density of generated DAG (#54) Closes #44 --- manm_cs/graph/graph_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/manm_cs/graph/graph_builder.py b/manm_cs/graph/graph_builder.py index dd0e544..a20646d 100644 --- a/manm_cs/graph/graph_builder.py +++ b/manm_cs/graph/graph_builder.py @@ -186,7 +186,7 @@ def generate_continuous_variable(self, parents, node_idx) -> 'ContinuousVariable def generate_dag(self, seed: int) -> 'DiGraph': # Generate graph using networkx package - G = nx.gnp_random_graph(n=self.num_nodes, p=self.edge_density, seed=seed, directed=True) + G = nx.gnp_random_graph(n=self.num_nodes, p=self.edge_density, seed=seed, directed=False) # Convert generated graph to DAG dag = nx.DiGraph() dag.add_nodes_from(G)