def summarize(sentences, k=5): filtered_sentences = filter_sentences(sentences) S = build_similarity_matrix(filtered_sentences) ranks = pagerank(S) return get_topk_sentences_generate(ranks, sentences, k)
def extract_keywords(sentences, k=5): filtered_sentences = filter_sentences(sentences, lowercase=False, stem=False) word_to_ix, ix_to_word = build_vocabulary(filtered_sentences) S = build_coo_matrix(filtered_sentences, word_to_ix) ranks = pagerank(S) return get_topk_keywords(ranks, ix_to_word, k)