Skip to content

Change aggregation strategy in NER pipeline example#1180

Open
Ujjwal-Sharma-Data wants to merge 1 commit intohuggingface:mainfrom
Ujjwal-Sharma-Data:patch-1
Open

Change aggregation strategy in NER pipeline example#1180
Ujjwal-Sharma-Data wants to merge 1 commit intohuggingface:mainfrom
Ujjwal-Sharma-Data:patch-1

Conversation

@Ujjwal-Sharma-Data
Copy link

Issue: TypeError: TokenClassificationPipeline._sanitize_parameters() got an unexpected keyword argument 'grouped_entities'

This error occurred because grouped_entities=True was incorrectly passed either to the ner() call or during pipeline initialization. For the transformers library's NER pipeline, entity grouping is configured at initialization using the aggregation_strategy parameter.

Solution: The fix involves initializing the NER pipeline with aggregation_strategy="simple".

Corrected the Named Entity Recognition (NER) pipeline initialization to use aggregation_strategy="simple" instead of passing grouped_entities=True to the call, resolving a TypeError and ensuring proper entity grouping.
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

Comments