Skip to content

helpers

helpers

Functions:

Name Description
entities_to_html

Visualize detected entities as HTML

entities_to_html(text, entities)

Visualize detected entities as HTML text: raw text containing entities entities: a list of NERPrediction objects

Returns: an HTML string

Source code in src/nemo_safe_synthesizer/pii_replacer/ner/helpers.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
def entities_to_html(text: str, entities: list):
    """Visualize detected entities as HTML
    text: raw text containing entities
    entities: a list of ``NERPrediction`` objects

    Returns: an HTML string
    """
    palette = [
        "#7aecec",
        "#bfeeb7",
        "#feca74",
        "#ff9561",
        "#aa9cfc",
        "#c887fb",
        "#9cc9cc",
        "#ffeb80",
        "#ff8197",
        "#ff8197",
        "#f0d0ff",
        "#bfe1d9",
        "#e4e7d2",
    ]

    colors = {}
    for index, name in enumerate(Entity):
        colors[name.name] = palette[index % len(palette)]

    options = {"ents": colors.keys(), "colors": colors}

    # Manually create display object in SpaCy format
    ents = []

    for x in sorted(entities, key=lambda i: i.start):
        ents.append({"start": x.start, "end": x.end, "label": x.label})

    ex = [{"text": text, "ents": ents, "title": None}]
    html = spacy.displacy.render(ex, style="ent", options=options, manual=True)
    return html