10a2ef19ee3851f2bd5aa6bc1b9d9cee3b03e5f4,src/spn/structure/leaves/histogram/Gradients.py,,histogramm_gradient,#Any#Any#Any#,9

Before Change

    breaks = node.breaks

    nd = data[:, node.scope[0]]
    marg_ids = np.isnan(nd)

    locs = np.searchsorted(breaks, nd)

    probs_left = histogram_ll(node.breaks, np.array(node.densities), locs - 1)
    probs_center = histogram_ll(node.breaks, np.array(node.densities), locs)
    probs_right = histogram_ll(node.breaks, np.array(node.densities), locs + 1)

    gradients = ((probs_center - probs_left) + probs_right - probs_center) / 2
    gradients[marg_ids] = np.nan

    return gradients.reshape((-1, 1))

After Change

    data = input_vals

    breaks = node.breaks
    gradient = np.full(input_vals.shape, np.nan)

    nd = data[:, node.scope[0]]

    locs = np.searchsorted(breaks, nd)

    probs_left = histogram_ll(node.breaks, np.array(node.densities), locs - 1)
    probs_center = histogram_ll(node.breaks, np.array(node.densities), locs)
    probs_right = histogram_ll(node.breaks, np.array(node.densities), locs + 1)

    gradient[:, node.scope] = (((probs_center - probs_left) + probs_right - probs_center) / 2).reshape(-1, 1)

    return gradient

Instances