Not sure why my Python code that uses Spacy to add a phone_number entity is not working

Question

The pattern works with matcher. But not as an entity? Here is my code:

import spacy
from spacy.pipeline import EntityRuler

nlp = spacy.load("en_core_web_sm")

patterns = [
    {
        "label": "PHONE_NUMBER",
        "pattern": [
            {"ORTH": "("},
            {"SHAPE": "ddd"},
            {"ORTH": ")"},
            {"IS_SPACE": True, "OP": "?"},
            {"SHAPE": "ddd"},
            {"ORTH": "-"},
            {"SHAPE": "dddd"},
        ],
    }
]

entity_ruler = EntityRuler(nlp, patterns=patterns, overwrite_ents=True)
nlp.add_pipe("entity_ruler", before="ner")

doc = nlp("You can reach me at (111) 111-1111.")

for ent in doc.ents:
    print(ent.text, ent.label_)

This returns:

111 CARDINAL
111 CARDINAL

Advice/help needed and appreciated. Thank you.

Nick Tallant · Accepted Answer · 2023-05-02 15:03:32Z

1

The problem was two seperate components—one constructed with the class EntityRuler and one constructed with nlp.add_pipe. The component created with the add_pipe method wasn't aware of your patterns. Using just one method and then adding the patterns to that component did the trick.

import spacy

nlp = spacy.load("en_core_web_sm")
patterns = [
    {
        "label": "PHONE_NUMBER",
        "pattern": [
            {"ORTH": "("},
            {"SHAPE": "ddd"},
            {"ORTH": ")"},
            {"IS_SPACE": True, "OP": "?"},
            {"SHAPE": "ddd"},
            {"ORTH": "-"},
            {"SHAPE": "dddd"},
        ],
    }
]

ruler = nlp.add_pipe("entity_ruler", before="ner")
ruler.add_patterns(patterns)

doc = nlp("You can reach me at (111) 111-1111.")

for ent in doc.ents:
    print(ent.text, ent.label_)

(111) 111-1111 PHONE_NUMBER

I read about the different ways to initialize the component here: https://spacy.io/api/entityruler

answered May 2, 2023 at 15:03

Nick Tallant

3253 silver badges6 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user1255603 Over a year ago

Thank you. I really appreciate your kind response. It works. Going back to the docs to see why I missed that (because of course now it seems obvious...).

Collectives™ on Stack Overflow

Not sure why my Python code that uses Spacy to add a phone_number entity is not working

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related