bricks icon indicating copy to clipboard operation
bricks copied to clipboard

[MODULE] - BERT NER extraction

Open LeonardPuettmannKern opened this issue 2 years ago • 2 comments

Please describe the module you would like to add to bricks We have some language models that we use for keyword extraction, but it would also be great to use these encoder models for keyword extraction. Starting with a "general" keyword extraction brick and perhaps more specific models in the future.

Do you already have an implementation?

Additional context The models from ML6 are really great for this: https://huggingface.co/ml6team/keyphrase-extraction-distilbert-kptimes

LeonardPuettmannKern avatar Jun 01 '23 09:06 LeonardPuettmannKern

Added with #317

LeonardPuettmannKern avatar Jul 21 '23 07:07 LeonardPuettmannKern

The output of the except statement needs to be changed, as it does not conform with the structure of refinery.

LeonardPuettmannKern avatar Sep 28 '23 12:09 LeonardPuettmannKern