Ensure reliable edge density of generated DAG (#54)

Closes #44
hpi-epic · Feb 9, 2023 · 9a09b49 · 9a09b49
1 parent 7b29d25
commit 9a09b49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)