diff options
Diffstat (limited to 'beliefs/factors/bernoulli_or_cpd.py')
-rw-r--r-- | beliefs/factors/bernoulli_or_cpd.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/beliefs/factors/bernoulli_or_cpd.py b/beliefs/factors/bernoulli_or_cpd.py index 12ee2f6..6e01cf9 100644 --- a/beliefs/factors/bernoulli_or_cpd.py +++ b/beliefs/factors/bernoulli_or_cpd.py @@ -20,7 +20,8 @@ class BernoulliOrCPD(TabularCPD): variable_card=2, parents=parents, parents_card=[2]*len(parents), - values=[]) + values=[], + state_names={var: ['False', 'True'] for var in [variable] + parents}) self._values = None @property |