def extract_entities()

in sagemaker_notebook_instance/containers/entity_recognition/entry_point.py [0:0]

• 11 lines of code
• 2 McCabe index (conditional complexity)

def extract_entities(spacy_document):
    entities = []
    for spacy_entity in spacy_document.ents:
        entity = {
            'text': spacy_entity.text,
            'start_char': spacy_entity.start_char,
            'end_char': spacy_entity.end_char,
            'label': spacy_entity.label_
        }
        entities.append(entity)
    return entities