diff --git a/cdt/data/acyclic_graph_generator.py b/cdt/data/acyclic_graph_generator.py index a4e8af0..6b6bcd1 100644 --- a/cdt/data/acyclic_graph_generator.py +++ b/cdt/data/acyclic_graph_generator.py @@ -111,7 +111,7 @@ def init_dag(self, verbose): self.adjacency_matrix[i, j] = 1 elif self.dag_type == 'erdos': - nb_edges = self.expected_degree * self.nodes + nb_edges = (self.expected_degree * self.nodes)/2 prob_connection = 2 * nb_edges/(self.nodes**2 - self.nodes) causal_order = np.random.permutation(np.arange(self.nodes))