search

code-kern-ai / bricks

Open-source natural language enrichments at your fingertips.
Apache License 2.0
448 stars 22 forks source link

[MODULE] - BERT NER extraction #313

Open LeonardPuettmannKern opened 1 year ago

LeonardPuettmannKern commented 1 year ago

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 commented 1 year ago

Added with #317

LeonardPuettmannKern commented 11 months ago

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