madisonmay / finetune Goto Github PK

Is your feature request related to a problem? Please describe.
In NER (named entity detection) we sometimes already know the segmentation of the entities, but still need to classify their type. E.g. in the sentece
'Paxar Corp said it has acquired Thermo-Print GmbH'
We might know that 'Paxar Corp' and 'Thermo-Print GmbH' are the relevant entities, but we want to predict their label as ORG.

Describe the solution you'd like
Perhaps add an optional param named spans to SequenceLabeler.predict, which is a list of dictionaries. Each dictionary will contain the start and end indices.