@REgorion

Проблема перевода с Python2 на Python3. В чём ошибка?

Код взял отсюда: 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: ('Русском', 'языке.')


В чём заключается ошибка и как её исправить?
  • Вопрос задан
  • 163 просмотра
Пригласить эксперта
Ответы на вопрос 1
Вся ошибка судя по всему в том, что текст оканчивается на точку.
дорогу выходившей С
Ответ написан
Комментировать
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы