Garmelon/trumpinator
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve style (based on Xyzzy's suggestions)

Browse source
This commit is contained in:
Joscha 2016-09-18 20:46:16 +00:00
parent aad6164d91
commit 932da577dd
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

14
markov.py
View file

@ -15,17 +15,17 @@ class Markov:
before = "" before = ""
for char in line: for char in line:
if not before in self.matrix: self.matrix[before] = {} if before not in self.matrix: self.matrix[before] = {}
if not char in self.matrix[before]: self.matrix[before][char] = 0 if char not in self.matrix[before]: self.matrix[before][char] = 0
self.matrix[before][char] += 1 self.matrix[before][char] += 1
before = (before + char)[-self.before:] before = (before + char)[-self.before:]
if not before in self.matrix: self.matrix[before] = {} if before not in self.matrix: self.matrix[before] = {}
if not "" in self.matrix[before]: self.matrix[before][""] = 0 if "" not in self.matrix[before]: self.matrix[before][""] = 0
self.matrix[before][""] += 1 self.matrix[before][""] += 1
def hiccup(self, before): def hiccup(self, before):
if not before in self.matrix: if before not in self.matrix:
before = random.choice(self.matrix.keys()) before = random.choice(self.matrix.keys())
selection = self.matrix[before] selection = self.matrix[before]
@ -71,12 +71,12 @@ def main(filename, paragraphs, char_lookback):
if __name__ == "__main__": if __name__ == "__main__":
try: try:
paragraphs = int(sys.argv[1]) paragraphs = int(sys.argv[1])
except: except (IndexError, ValueError):
paragraphs = 1 paragraphs = 1
try: try:
char_lookback = int(sys.argv[2]) char_lookback = int(sys.argv[2])
except: except (IndexError, ValueError):
char_lookback = 10 char_lookback = 10
main("trump_speech.txt", paragraphs, char_lookback) main("trump_speech.txt", paragraphs, char_lookback)