KeyBERT
KeyBERT copied to clipboard
keybert: Fine tuning custom data with few examples
Hi,
@MaartenGr @koaning @artmatsak @kunihik0 Is there a way to fine-tune keybert with custom labelled data to generate meaningful keywords? Please refer me a way to fine-tune keyword extraction on custom data. Thanks in advance.
Thanks in advance.
Fine-tuning is not possible within KeyBERT as it uses pre-trained models for semantic similarity. The pre-trained models can all differ in their architecture as well as their underlying libraries. Having said that, you can fine-tune those models and pass them to KeyBERT. Typically, this is typically a good place to start training a model.
Due to inactivity, I'll be closing this for now. However, if you want to continue this discussion or if something is not clear, please let me know!