Markov: many working backwards bugfixes wrapped together

This commit is contained in:
Brian S. Stephan 2012-07-29 22:36:11 -05:00
parent bf850592df
commit e8e4354358
1 changed files with 14 additions and 4 deletions

View File

@ -369,13 +369,23 @@ class Markov(Module):
# now work backwards until we randomly bump into a start
while True:
key_hits = self._retrieve_k2_for_value(working_backwards[0], context_id)
if self.start2 in key_hits:
self.log.debug("working backwards found start, finishing")
gen_words = gen_words + working_backwards
if self.start2 in key_hits and len(working_backwards) > 2:
self.log.debug("working backwards forced start, finishing")
gen_words += working_backwards
self.log.debug("gen_words: {0:s}".format(" ".join(gen_words)))
break
elif len(key_hits) == 0:
self.log.debug("no key_hits, finishing")
gen_words += working_backwards
self.log.debug("gen_words: {0:s}".format(" ".join(gen_words)))
break
elif len(key_hits) == 1 and self.start2 in key_hits:
self.log.debug("only start2 in key_hits, finishing")
gen_words += working_backwards
self.log.debug("gen_words: {0:s}".format(" ".join(gen_words)))
break
else:
found_word = random.choice(filter(lambda a: a != self.stop, key_hits))
found_word = random.choice(filter(lambda a: a != self.start2, key_hits))
working_backwards.insert(0, found_word)
self.log.debug("added '{0:s}' to working_backwards".format(found_word))
self.log.debug("working_backwards: {0:s}".format(" ".join(working_backwards)))