It would be a great feature to support entity specific thresholds.
This helps with models and entities where the threshold is skewed due to different pre-training efforts.
We probably have to do multi label prediction under the hood and then filter the results per entity threshold.
Config should probably be done as a dict of entity_label: threshold, but we still want to support one overall threshold and entities using it.