python - 是否有纠正大小写的python拼写纠正库？

Question

我有拼写正确但全部小写(第一个字符除外)的字符串，我想纠正大写(英语 - 所以基本上只是事物的名称......)。我尝试了 pyspellcheck、autocorrect 和 symspellpy，它们不考虑大写 afaik。

因此，例如字符串 'And then we went to see frank from england to have a beer with him.' 应该更正为 'And then we went to see Frank from England和他一起喝啤酒。”。

你知道有哪个图书馆可以做到这一点吗？

Answer 1

你可以用 spaCy 做到这一点:

import spacy
nlp=spacy.load('en_core_web_md')


def capitalize_ent(text):
    title_text=text.title()
    print(text)
    doc=nlp(title_text)
    words=[]
    for x in doc:
        if nlp(x.text).ents:
            words.append(x.text)
    for word in words:
        text=text.replace(word.lower(),word)
    return text
            

不要忘记下载 spaCy 语言模型:

python -m spacy download en_core_web_md