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

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.brown.words(categories = ['religion','science_fiction'])
bigrams = nltk.bigrams(text)
cfd = nltk.ConditionalFreqDist(bigrams)

generate_model(cfd, 'but')