0a0688c0fd83165b428d1eafb1b6fb19e1674e29,theanolm/src/theanolm/dictionary.py,WordClass,sample,#WordClass#,71

Before Change

            :returns: a random word from this class
            

            return next(iter(self._probs.keys()))

    def __init__(self, input_file, input_format):
        Creates word classes.

After Change

            

            words = list(self._probs.keys())
            probs = list(self._probs.values())
            sample_distribution = numpy.random.multinomial(1, probs)
            indices = numpy.flatnonzero(sample_distribution)
            assert len(indices) == 1
            return words[indices[0]]

    def __init__(self, input_file, input_format):
        Creates word classes.

Instances