Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Choosing spaCy model #8

Open
omstuhler opened this issue May 20, 2022 · 1 comment
Open

Choosing spaCy model #8

omstuhler opened this issue May 20, 2022 · 1 comment

Comments

@omstuhler
Copy link

Thank you for this fantastic library!

Which spacy model is used when running booknlp.process()? Is this in any way controlled by the "model" parameter ("small", "big") or does it simply use the model that is currently initialized. For example, could I get it to use the en_core_web_trf by running the following before I use booknlp.process():

import spacy
nlp = spacy.load('en_core_web_trf')
@lucy3
Copy link

lucy3 commented Apr 24, 2023

The default is en_core_web_sm (line 32, here)

You can specify the spacy model in the model params you pass into booknlp:

    model_params = {
        "pipeline": "entity,quote,supersense,event,coref",
        "model": "big",
        "spacy_model": "en_core_web_trf",
    }

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants