diff options
Diffstat (limited to 'beliefs/models/DirectedGraph.py')
-rw-r--r-- | beliefs/models/DirectedGraph.py | 35 |
1 files changed, 0 insertions, 35 deletions
diff --git a/beliefs/models/DirectedGraph.py b/beliefs/models/DirectedGraph.py deleted file mode 100644 index 8dfb9bd..0000000 --- a/beliefs/models/DirectedGraph.py +++ /dev/null @@ -1,35 +0,0 @@ -import networkx as nx - - -class DirectedGraph(nx.DiGraph): - """ - Base class for all directed graphical models. - """ - def __init__(self, edges, node_labels): - """ - Input: - edges: an edge list, e.g. [(parent1, child1), (parent1, child2)] - node_labels: a list of strings of node labels - """ - super().__init__() - if edges is not None: - self.add_edges_from(edges) - if node_labels is not None: - self.add_nodes_from(node_labels) - - def get_leaves(self): - """ - Returns a list of leaves of the graph. - """ - return [node for node, out_degree in self.out_degree_iter() if - out_degree == 0] - - def get_roots(self): - """ - Returns a list of roots of the graph. - """ - return [node for node, in_degree in self.in_degree().items() if - in_degree == 0] - - def get_topologically_sorted_nodes(self, reverse=False): - return nx.topological_sort(self, reverse=reverse) |