INNER CODE UNIT · Python

_initialize_classification_model

dragondrop-cloud/cloud-concierge · nlpengine/main.py:178

def _initialize_classification_model(nlp: spacy.Language, initial_examples) -> Union:
    """Initialize the multi-label classification model."""
    config = {
        "threshold": 0.5,
        "model": DEFAULT_MULTI_TEXTCAT_MODEL,
    }

    textcat_multilabel = nlp.add_pipe("textcat_multilabel", config=config)

    textcat_multilabel.initialize(lambda: initial_examples, nlp=nlp)

    return textcat_multilabel


def _train_model(
    nlp: spacy.Language,
    textcat_multilabel_model,
    training_data: List[Example],

View source record →