diff options
author | Cathy Yeh <cathy@driver.xyz> | 2017-12-03 20:38:28 -0800 |
---|---|---|
committer | Cathy Yeh <cathy@driver.xyz> | 2017-12-03 20:38:28 -0800 |
commit | 26b43410569044aff46053cae7c68862825dd4ec (patch) | |
tree | b184df84d416e2ddf837b25baadff4f9feaaa250 /beliefs/models/DirectedGraph.py | |
parent | 6a1b35f5bf122232d058ed0f3ea19c15629c0cbc (diff) | |
parent | c906bd37fba63ba706cc3b7802bfb18ffb05ee9a (diff) | |
download | beliefs-26b43410569044aff46053cae7c68862825dd4ec.tar.gz beliefs-26b43410569044aff46053cae7c68862825dd4ec.tar.bz2 beliefs-26b43410569044aff46053cae7c68862825dd4ec.zip |
LGS-164 belief propagation for polytrees, special case of OR cpds, refactored from LGSv0.0.2
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) |