diff options
| author | Cathy Yeh <cathy@driver.xyz> | 2017-11-20 11:40:02 -0800 |
|---|---|---|
| committer | Cathy Yeh <cathy@driver.xyz> | 2017-11-20 11:40:02 -0800 |
| commit | 71e384a741e52f94882b14062a3dc10e5f391533 (patch) | |
| tree | 669b8c78e3c9c7e44cf58692fef81836b8cc94b9 /beliefs/inference/belief_propagation.py | |
| parent | b16e990b7e4d00e427d4445ba38eef0fb967963a (diff) | |
| download | beliefs-71e384a741e52f94882b14062a3dc10e5f391533.tar.gz beliefs-71e384a741e52f94882b14062a3dc10e5f391533.tar.bz2 beliefs-71e384a741e52f94882b14062a3dc10e5f391533.zip | |
BernoulliOrCPD inherits from TabularCPD
Diffstat (limited to 'beliefs/inference/belief_propagation.py')
| -rw-r--r-- | beliefs/inference/belief_propagation.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/beliefs/inference/belief_propagation.py b/beliefs/inference/belief_propagation.py index ecd5e9c..37aa437 100644 --- a/beliefs/inference/belief_propagation.py +++ b/beliefs/inference/belief_propagation.py @@ -54,8 +54,8 @@ class BeliefPropagation: # exclude the message sender (either a parent or child) from getting an # outgoing msg from the node to update - parent_ids = node.parents - set([msg_sender_label_id]) - child_ids = node.children - set([msg_sender_label_id]) + parent_ids = set(node.parents) - set([msg_sender_label_id]) + child_ids = set(node.children) - set([msg_sender_label_id]) print("parent_ids:", parent_ids) print("child_ids:", child_ids) |