def main(sentence_number): print "\n\n---UNIGRAM SENTENCES---\n" generate_unigram_sentences(unigram_model, sentence_number) print "\n\n---BIGRAM SENTENCES---\n" generate_bigram_sentences(bigram_model, sentence_number) print "\n\n---TRIGRAM SENTENCES---\n" generate_trigram_sentences(trigram_model, sentence_number)
def main(seed_value, gen_num_sentences): # tokenize the supplied sentence fragment seed_tokens = word_tokenize(seed_value) # create "N" random sentences, based on sentence fragment "seed" if len(seed_tokens) > 0: for i in range(gen_num_sentences): generate_bigram_sentences(bigram_model, 1, seed_value, seed_tokens[-1]) else: # if sentence fragment missing, just generate from a random word in corpus for i in range(gen_num_sentences): generate_bigram_sentences(bigram_model, 1, seed_value, ".")