Example #1
0
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)
Example #2
0
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, ".")