kewpie

KEyWord PIckEr with tf-idf


Keywords
keyword-extraction, machine-learning, python, tf-idf
License
MIT
Install
pip install kewpie==0.0.2

Documentation

kewpie

KEyWord PIckEr with tf-idf

Work in Progress.

Installation

pip install kewpie

Basic Usage

from kewpie import KwPicker

docs = [
    "I am a cat.",
    "I am a dog."
]

picker = KwPicker.build(docs, savedir='/dir/to/save/model/')

sentence = "I am a cat."
span, keyword = picker.get_keywords(sentence)

print(keyword)  # -> 'cat'