是否可以更改Spacy令牌生成器的令牌拆分规则?

时间:2019-07-31 17:16:06

标签: spacy

(德语)spacy标记生成器默认情况下不会在斜线,下划线或星号上分割,这正是我所需要的(因此,“ der / die”生成单个标记)。

但是它确实在括号上分开,因此“ dies(und)das”被分成5个令牌。是否有一种(简单的)方式告诉默认令牌生成器也不要在括号内进行拆分,括号内的字母两侧都没有空格?

为令牌生成器定义的括号上的拆分到底是什么?

1 个答案:

答案 0 :(得分:1)

在此行中定义了括号的拆分,在括号中的拆分在两个字母之间:

https://github.com/explosion/spaCy/blob/23ec07debdd568f09c7c83b10564850f9fa67ad4/spacy/lang/de/punctuation.py#L18

没有简单的方法来删除中缀模式,但是您可以定义一个自定义标记器来实现您想要的功能。一种方法是从spacy/lang/de/punctuation.py复制中缀定义并进行修改:

import re
import spacy
from spacy.tokenizer import Tokenizer
from spacy.lang.char_classes import ALPHA, ALPHA_LOWER, ALPHA_UPPER, CONCAT_QUOTES, LIST_ELLIPSES, LIST_ICONS
from spacy.lang.de.punctuation import _quotes
from spacy.util import compile_prefix_regex, compile_infix_regex, compile_suffix_regex

def custom_tokenizer(nlp):
    infixes = (
        LIST_ELLIPSES
        + LIST_ICONS
        + [
            r"(?<=[{al}])\.(?=[{au}])".format(al=ALPHA_LOWER, au=ALPHA_UPPER),
            r"(?<=[{a}])[,!?](?=[{a}])".format(a=ALPHA),
            r'(?<=[{a}])[:<>=](?=[{a}])'.format(a=ALPHA),
            r"(?<=[{a}]),(?=[{a}])".format(a=ALPHA),
            r"(?<=[{a}])([{q}\]\[])(?=[{a}])".format(a=ALPHA, q=_quotes),
            r"(?<=[{a}])--(?=[{a}])".format(a=ALPHA),
            r"(?<=[0-9])-(?=[0-9])",
        ]
    )

    infix_re = compile_infix_regex(infixes)

    return Tokenizer(nlp.vocab, prefix_search=nlp.tokenizer.prefix_search,
                                suffix_search=nlp.tokenizer.suffix_search,
                                infix_finditer=infix_re.finditer,
                                token_match=nlp.tokenizer.token_match,
                                rules=nlp.Defaults.tokenizer_exceptions)


nlp = spacy.load('de')
nlp.tokenizer = custom_tokenizer(nlp)