diff options
author | Cathy Yeh <cathy@driver.xyz> | 2017-11-20 17:05:37 -0800 |
---|---|---|
committer | Cathy Yeh <cathy@driver.xyz> | 2017-11-21 13:18:34 -0800 |
commit | d166e36eaf5803af035e444628c67701322b0eb6 (patch) | |
tree | 3e715d2ab34ce447222ccfa11bcde31065faae26 /beliefs/inference | |
parent | 71e384a741e52f94882b14062a3dc10e5f391533 (diff) | |
download | beliefs-d166e36eaf5803af035e444628c67701322b0eb6.tar.gz beliefs-d166e36eaf5803af035e444628c67701322b0eb6.tar.bz2 beliefs-d166e36eaf5803af035e444628c67701322b0eb6.zip |
refactor msg passing methods to BeliefUpdateNodeModel from BayesianModel
Diffstat (limited to 'beliefs/inference')
-rw-r--r-- | beliefs/inference/belief_propagation.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/beliefs/inference/belief_propagation.py b/beliefs/inference/belief_propagation.py index 37aa437..02f5595 100644 --- a/beliefs/inference/belief_propagation.py +++ b/beliefs/inference/belief_propagation.py @@ -1,7 +1,8 @@ import numpy as np from collections import namedtuple -from beliefs.types.Node import InvalidLambdaMsgToParent +from beliefs.models.beliefupdate.Node import InvalidLambdaMsgToParent +from beliefs.models.beliefupdate.BeliefUpdateNodeModel import BeliefUpdateNodeModel from beliefs.utils.math_helper import is_kronecker_delta @@ -22,10 +23,12 @@ class BeliefPropagation: def __init__(self, model, inplace=True): """ Input: - model: an instance of BayesianModel class or subclass + model: an instance of BeliefUpdateNodeModel inplace: bool modify in-place the nodes in the model during belief propagation """ + if not isinstance(model, BeliefUpdateNodeModel): + raise TypeError("Model must be an instance of BeliefUpdateNodeModel") if inplace is False: self.model = model.copy() else: |