def __next__(self): if len(self._start_words) > 0: next_word = self._start_words[0] self._start_words = self._start_words[1:] return next_word while True: try: next_word = discrete_sample(self.chain[self._prev_words]) self._prev_words = self._prev_words[1:] + (next_word,) return next_word except: raise StopIteration()
def __next__(self): if len(self._start_words) > 0: next_word = self._start_words[0] self._start_words = self._start_words[1:] return next_word while True: try: next_word = discrete_sample(self.chain[self._prev_words]) self._prev_words = self._prev_words[1:] + (next_word, ) return next_word except: raise StopIteration()
def __iter__(self): self._start_words = discrete_sample(self.capitals) self._prev_words = self._start_words return self