178553d4d4edfdc155d876c5ac75aaa1105445ee,src/spn/structure/leaves/conditional/utils.py,,get_scipy_obj_params,#Any#Any#,11

Before Change

    elif isinstance(node, Conditional_Bernoulli):
        // assert len(node.p) == obs.shape[0]
        p = node.inv_linkfunc(np.dot(obs, node.weights))
        assert len(p) == obs.shape[0]
        return bernoulli, {"p": p}

    else:

After Change

    pred = np.dot(obs, node.weights[:-1]) + node.weights[-1]
    if isinstance(node, Conditional_Gaussian):
        mean = pred
        return norm, {"loc": mean, "scale": np.ones(obs.shape[0])}

    elif isinstance(node, Conditional_Poisson):
        mu = np.exp(pred)

Instances