KeyBERT icon indicating copy to clipboard operation
KeyBERT copied to clipboard

keybert: Fine tuning custom data with few examples

Open gauravgund opened this issue 2 years ago • 1 comments

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.

gauravgund avatar Jul 29 '22 05:07 gauravgund

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.

MaartenGr avatar Jul 31 '22 05:07 MaartenGr

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!

MaartenGr avatar Nov 03 '22 08:11 MaartenGr