NLTk Ex 2.24.a Алгоритм порождения случайного связного текста

gistfile1.py

#Суть алгоритма такова: из текста мы делаем множество биграмм - кортежей вида (x,y).
#Затем мы берем начальное слово и выбираем другое случайное, идущее с ним в биграмме.
#К этому случайному добавляем очередное случайное из биграммы и т.л.
#Случайность выбора помогает не создавать "петлей"

import nltk
import random

def generate_model(cfdist, word, num=15):
    for i in range(num):
        print word,
        word = random.choice(list(cfdist[word]))

text = nltk.corpus.genesis.words('english-kjv.txt') #берем текст
bigrams = nltk.bigrams(text) #разбиваем на биграммы
cfd = nltk.ConditionalFreqDist(bigrams) #в принципе можно обойтись без этого

generate_model(cfd, 'living')