1187f93b53b14ed61704c0f07d618c47822f180a,gpytorch/random_variables/categorical_random_variable.py,CategoricalRandomVariable,sample,#CategoricalRandomVariable#,24

Before Change

        p = random.random()
        cmf_lt = self._cumulative_mass_function.ge(p)
        for i,v in enumerate(cmf_lt):
            if v == 1:
                return torch.LongTensor([i])

    def num_categories(self):
        return len(self.mass_function)

After Change

        return math.log(self.mass_function[i])

    def sample(self):
        return torch.multinomial(self.mass_function, 1)

    def num_categories(self):
        return len(self.mass_function)

Instances