Код взял отсюда:
agiliq.com/blog/2009/06/generating-pseudo-random-t...
Немного его исправил(xrange поменял на range):
import random
class Markov(object):
def __init__(self, open_file):
self.cache = {}
self.open_file = open_file
self.words = self.file_to_words()
self.word_size = len(self.words)
self.database()
def file_to_words(self):
self.open_file.seek(0)
data = self.open_file.read()
words = data.split()
return words
def triples(self):
if len(self.words) < 3:
return
for i in range(len(self.words) - 2):
yield (self.words[i], self.words[i+1], self.words[i+2])
def database(self):
for w1, w2, w3 in self.triples():
key = (w1, w2)
if key in self.cache:
self.cache[key].append(w3)
else:
self.cache[key] = [w3]
def generate_markov_text(self, size=25):
seed = random.randint(0, self.word_size-3)
seed_word, next_word = self.words[seed], self.words[seed+1]
w1, w2 = seed_word, next_word
gen_words = []
for i in range(size):
gen_words.append(w1)
w1 , w2 = w2, random.choice(self.cache[(w1, w2)])
gen_words.append(w2)
return ' '.join(gen_words)
Код вызова:
file_ = open('text.txt')
import markovgen
markov = markovgen.Markov(file_)
markov.generate_markov_text()
text.txt:
Простой текст, который ничего не значит. А ещё он написан на Русском языке.
После выполнение выдаёт мне следующую ошибку:
line 42, in generate_markov_text
w1 , w2 = w2, random.choice(self.cache[(w1, w2)])
KeyError: ('Русском', 'языке.')
В чём заключается ошибка и как её исправить?
