46324d0e05c679b2cf50b81289fe2886456f6e51,dit/pid/ibroja.py,BROJAOptimizer,objective,#BROJAOptimizer#Any#,45

Before Change

        pmf = self._expand(x).reshape(self._shape)
        h_total = -np.nansum(pmf * np.log2(pmf))

        inputs = tuple(range(len(self._shape) - 1))
        p_output = pmf.sum(axis=inputs)
        h_output = -np.nansum(p_output * np.log2(p_output))

        input_pmf = pmf.sum(axis=-1)

After Change

        reduced_pmf = pmf.sum(axis=0)
        h_reduced = -np.nansum(reduced_pmf * np.log2(reduced_pmf))

        others_pmf = pmf.sum(axis=(0, 2))
        h_others = -np.nansum(others_pmf * np.log2(others_pmf))

        omi = h_others + h_output - h_reduced

Instances