Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- named_entities = []
- for sentence in corpus:
- temp_entity_name = ''
- temp_named_entity = None
- sentence = nlp(sentence)
- for word in sentence:
- term = word.text
- tag = word.ent_type_
- if tag:
- temp_entity_name = ' '.join([temp_entity_name, term]).strip()
- temp_named_entity = (temp_entity_name, tag)
- else:
- if temp_named_entity:
- named_entities.append(temp_named_entity)
- temp_entity_name = ''
- temp_named_entity = None
- entity_frame = pd.DataFrame(named_entities,
- columns=['Entity Name', 'Entity Type'])
Add Comment
Please, Sign In to add comment