Classy-Classification Example code gives error

[the-black-adder]

Hi,

When trying the first example I get an error. I have downloaded en_core_web_lg and other models and I can't find a fix for this.

Config validation error
classy_classification -> model: extra fields not permitted
{'nlp': <spacy.lang.en.English object at 0x0000019E780DE290>, 'name': 'classy_classification', 'cat_type': 'few', 'config': None, 'data': {'economy': ['The cost of living', 'Tax', 'Petrol prices', 'Inflation'], 'public trust': ['Breaking the rules', 'One rule for them', 'lied', "don't live in the real world"]}, 'device': 'cpu', 'include_doc': True, 'include_sent': False, 'model': None, 'model:': 'spacy', 'multi_label': False, 'verbose': False, 'https://github.com/factories': 'classy_classification'}
python-BaseException

This my code using python v3.11 and spaCy v3.7.2, classy-classification 0.6.7.

import spacy
import classy_classification

# Dictionary containing categories (keys) and lists matched to each category (value).
data = {
"economy": [
"The cost of living",
"Tax",
"Petrol prices",
"Inflation"],
"public trust": [
"Breaking the rules",
"One rule for them",
"lied",
"don't live in the real world"
]

}

# Load model used for NLP
nlp = spacy.load("en_core_web_lg")

# Set up a pipeline to feed our data into a text categoriser
nlp.add_pipe(
"classy_classification",
config={
"data": data,
"model:": "spacy"
}
)

# Feed a sentence into the pipeline to get a score for each category in our dictionary
print(nlp("I don't trust them")._.cats)

Thanks.
Matt.

[rmitsch]

Hi @the-black-adder, we don't maintain classy-classification. I see you've brought up this issue on their board already, I recommend you follow up there.

Speaking generally the error indicates that "model" is an unexpected attribute in the initialization of your classy component (nlp.add_pipe(...)). I'd say look into this, maybe this has changed in between versions.