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
def __iter__(self):
self._start_words = discrete_sample(self.capitals)
self._prev_words = self._start_words
return self
