cf56f37bdd8a932ec2dbef6dd47ce4cfbb820018,pgmpy/inference/dbn_inference.py,DBNInference,forward_inference,#DBNInference#,162

Before Change

        for time in range(1, time_range + 1):
            evidence_time = self._get_evidence(evidence, time, 1)
            if interface_nodes_dict:
                if evidence_time:
                    evidence_time.update(interface_nodes_dict)
                else:
                    evidence_time = interface_nodes_dict

            if variable_dict[time]:
                variable_time = self._shift_nodes(variable_dict[time], 1)
                new_values = mid_bp.query(variable_time, evidence=evidence_time)
                changed_values = {}

After Change

            variable_dict[var[1]].append(var)
        time_range = max(variable_dict)
        if evidence:
            evid_time_range = max([time for var, time in evidence.keys()])
            time_range = max(time_range, evid_time_range)
        start_bp = BeliefPropagation(self.start_junction_tree)
        mid_bp = BeliefPropagation(self.one_and_half_junction_tree)
        evidence_0 = self._get_evidence(evidence, 0, 0)
        interface_nodes_dict = {}

Instances