46cb2800632916cd005a974f72c648a4524c08c2,pixyz/losses/mmd.py,MMD,_get_estimated_value,#MMD#Any#,47

Before Change

        return "MMD[{}||{}]".format(self._p.prob_text, self._q.prob_text)

    def _get_estimated_value(self, x={}, **kwargs):
        batch_size = get_dict_values(x, self.input_dist.input_var[0])[0].shape[0]

        // sample from distributions
        p_x = get_dict_values(self._p.sample(x, batch_size=batch_size), self._p.var)[0]
        q_x = get_dict_values(self._q.sample(x, batch_size=batch_size), self._q.var)[0]

After Change

        return get_dict_values(x, self.input_dist.input_var[0])[0].shape[0]

    def _get_estimated_value(self, x={}, **kwargs):
        batch_size = self._get_batch_size(x)

        // sample from distributions
        p_x = get_dict_values(self._p.sample(x, batch_size=batch_size), self._p.var)[0]
        q_x = get_dict_values(self._q.sample(x, batch_size=batch_size), self._q.var)[0]

Instances