Adding models

[louisdo]

Hi @MaartenGr,

Thank you for your efforts towards KeyBERT. I'm wondering is it possible to add two of my own models to KeyBERT

• ERU-KG
• CASPER

The former is a keyphrase generation model, while the latter is a sparse lexicon-based text embedding build for retrieval. Both can be used for keyphrase generation/extraction. The strength of these models is that they are very fast, as they do not require modeling noun phrase candidates (do not need to produce embeddings for keyphrase candidates, just need to do so for the input text).

If it is possible to add models. I would love to contribute. Please let me know if this is possible.

[MaartenGr]

Hi! I'm definitely open to exploring adding models to KeyBERT. Do you have existing code lying somewhere that I can take a look at? It is important that whatever is added to KeyBERT fits within the philosophy and design of KeyBERT. More specifically, what kinds of changes would be needed in KeyBERT and how would that affect its API? If it is merely adding models that are completely independent of KeyBERT, then I wonder if that would fit here.