thallada/nlp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Try and fix occasional exception

Browse Source
This commit is contained in:
Tyler Hallada 2016-08-14 17:43:03 -04:00
parent 92e34b7b76
commit 3ace25b6e2
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
generate_poem.py
View File

@ -103,10 +103,14 @@ class PoemGenerator():
haiku = haiku + ' '.join(first) + '\n' haiku = haiku + ' '.join(first) + '\n'
next_words = [freq[0] for freq in self.cfd[first[-1]].items() next_words = [freq[0] for freq in self.cfd[first[-1]].items()
if not self.only_punctuation.match(freq[0])] if not self.only_punctuation.match(freq[0])]
if not next_words:
next_words = self.all_words
second = self.haiku_line([], 0, next_words, 7) second = self.haiku_line([], 0, next_words, 7)
haiku = haiku + ' '.join(second) + '\n' haiku = haiku + ' '.join(second) + '\n'
next_words = [freq[0] for freq in self.cfd[second[-1]].items() next_words = [freq[0] for freq in self.cfd[second[-1]].items()
if not self.only_punctuation.match(freq[0])] if not self.only_punctuation.match(freq[0])]
if not next_words:
next_words = self.all_words
third = self.haiku_line([], 0, next_words, 5) third = self.haiku_line([], 0, next_words, 5)
haiku = haiku + ' '.join(third) + '\n' haiku = haiku + ' '.join(third) + '\n'
return haiku return haiku