Python Keyphrase Extraction (pke) toolkit

pke is an open source python-based keyphrase extraction toolkit that provides an end-to-end keyphrase extraction pipeline in which each component can be easily modified or extented to develop new approaches.

Pre-trained models for keyphrase generation

bart-base-kp20k is a bart-base model for scientific keyphrase generation fine-tuned on the KP20k dataset.

bart-large-kp20k is a bart-large model for scientific keyphrase generation fine-tuned on the KP20k dataset.