7281b68353fa8f9263e071716ac5081df5168f59,pgmpy/inference/ExactInference.py,VariableElimination,_variable_elimination,#VariableElimination#Any#Any#Any#Any#,9

Before Change

        for query_var in variables:
            phi = factor_product(*final_distribution)
            phi.marginalize(list(set(variables) - set([query_var])))
            if evidence:
                phi.reduce(["{evidence_var}_{evidence}".format(
                    evidence_var=evidence_var, evidence=evidence[evidence_var])
                    for evidence_var in evidence])
                phi.normalize()

            query_var_factor[query_var] = phi
        return query_var_factor

    def query(self, variables, evidence=None, elimination_order=None):

After Change

        // Dealing with evidence. Reducing factors over it before VE is run.
        for evidence_var in evidence:
            for factor in working_factors[evidence_var]:
                factor_reduced = factor.reduce("{evidence_var}_{state}".format(
                    evidence_var=evidence_var, state=evidence[evidence_var]), inplace=False).normalize()
                for var in factor_reduced.scope():
                    working_factors[var].remove(factor).add(factor_reduced)

        // TODO: Modify it to find the optimal elimination order

Instances