urchade / GLiNER

Generalist and Lightweight Model for Named Entity Recognition (Extract any entity types from texts) @ NAACL 2024
https://arxiv.org/abs/2311.08526
Apache License 2.0
1.48k stars 127 forks source link

Fix fine-tuning eval with prediction probabilities #47

Closed th0rntwig closed 8 months ago

th0rntwig commented 8 months ago

The addition of prediction scores causes ValueError: too many values to unpack (expected 3) during evaluation when fine-tuning. This PR solves this by handling predictions separately.

urchade commented 8 months ago

Thanks!