Adding models
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 4.2k
- Forks
- 385
- PR merge metrics
- No merged PRs in 30d
Description
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](https://arxiv.org/abs/2505.24219)
- [CASPER](https://arxiv.org/abs/2508.13394)
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.
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
The issue names no files, tests, or entry points. Start by reviewing how KeyBERT currently integrates models, then determine the requirements for ERU-KG and CASPER from the linked papers. Done means both models are supported for keyphrase extraction with appropriate test coverage.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 35/100