aboutsummaryrefslogtreecommitdiff
path: root/beliefs/types/BernoulliOrNode.py
diff options
context:
space:
mode:
authorCathy Yeh <cathy@driver.xyz>2017-11-20 11:40:02 -0800
committerCathy Yeh <cathy@driver.xyz>2017-11-20 11:40:02 -0800
commit71e384a741e52f94882b14062a3dc10e5f391533 (patch)
tree669b8c78e3c9c7e44cf58692fef81836b8cc94b9 /beliefs/types/BernoulliOrNode.py
parentb16e990b7e4d00e427d4445ba38eef0fb967963a (diff)
downloadbeliefs-71e384a741e52f94882b14062a3dc10e5f391533.tar.gz
beliefs-71e384a741e52f94882b14062a3dc10e5f391533.tar.bz2
beliefs-71e384a741e52f94882b14062a3dc10e5f391533.zip
BernoulliOrCPD inherits from TabularCPD
Diffstat (limited to 'beliefs/types/BernoulliOrNode.py')
-rw-r--r--beliefs/types/BernoulliOrNode.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/beliefs/types/BernoulliOrNode.py b/beliefs/types/BernoulliOrNode.py
index 27da85a..ce497b9 100644
--- a/beliefs/types/BernoulliOrNode.py
+++ b/beliefs/types/BernoulliOrNode.py
@@ -6,7 +6,7 @@ from beliefs.types.Node import (
MessageType,
InvalidLambdaMsgToParent
)
-from beliefs.factors.BernoulliOrFactor import BernoulliOrFactor
+from beliefs.factors.BernoulliOrCPD import BernoulliOrCPD
class BernoulliOrNode(Node):
@@ -18,7 +18,7 @@ class BernoulliOrNode(Node):
children=children,
parents=parents,
cardinality=2,
- cpd=BernoulliOrFactor(label_id, parents))
+ cpd=BernoulliOrCPD(label_id, parents))
def compute_pi_agg(self):
if not self.parents: