Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Lorenzo Ampil's avatar
    Allow the creation of "entity groups" for NerPipeline #3548 (#3957) · 18d233d5
    Lorenzo Ampil authored 
    * Add index to be returned by NerPipeline to allow for the creation of

* Add entity groups

* Convert entity list to dict

* Add entity to entity_group_disagg atfter updating entity gorups

* Change 'group' parameter to 'grouped_entities'

* Add unit tests for grouped NER pipeline case

* Correct variable name typo for NER_FINETUNED_MODELS

* Sync grouped tests to recent test updates
    18d233d5
test_pipelines.py 16.7 KB