diff options
Diffstat (limited to 'beliefs/types')
-rw-r--r-- | beliefs/types/BernoulliOrNode.py | 4 | ||||
-rw-r--r-- | beliefs/types/Node.py | 2 |
2 files changed, 3 insertions, 3 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: diff --git a/beliefs/types/Node.py b/beliefs/types/Node.py index a8dca7c..a496acf 100644 --- a/beliefs/types/Node.py +++ b/beliefs/types/Node.py @@ -33,7 +33,7 @@ class Node: parents: set of strings cardinality: int, cardinality of the random variable the node represents cpd: an instance of a conditional probability distribution, - e.g. BernoulliOrFactor or pgmpy's TabularCPD + e.g. BernoulliOrCPD or TabularCPD """ self.label_id = label_id self.children = children |