bricks
bricks copied to clipboard
[MODULE] - BERT NER extraction
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
Added with #317
The output of the except statement needs to be changed, as it does not conform with the structure of refinery.