annif.analyzer.spacy

Complexity

Total Complexity

5

Size/Duplication

Total Lines	37
Duplicated Lines	70.27 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	SpacyAnalyzer.__init__()	13	13	3
A	SpacyAnalyzer.tokenize_words()	9	9	2

"""spaCy analyzer for Annif which uses spaCy for lemmatization"""

import spacy
from . import analyzer
from annif.exception import OperationFailedException
import annif.util

_KEY_LOWERCASE = 'lowercase'


class SpacyAnalyzer(analyzer.Analyzer):

This code seems to be duplicated in your project.

    name = "spacy"

    def __init__(self, param, **kwargs):
        self.param = param
        try:
            self.nlp = spacy.load(param, exclude=['ner', 'parser'])
        except IOError as err:
            raise OperationFailedException(
                f"Loading spaCy model '{param}' failed - " +
                f"please download the model.\n{err}")
        if _KEY_LOWERCASE in kwargs:
            self.lowercase = annif.util.boolean(kwargs[_KEY_LOWERCASE])
        else:
            self.lowercase = False
        super().__init__(**kwargs)

    def tokenize_words(self, text, filter=True):
        lemmas = [lemma
                  for lemma in (token.lemma_
                                for token in self.nlp(text.strip()))
                  if (not filter or self.is_valid_token(lemma))]
        if self.lowercase:
            return [lemma.lower() for lemma in lemmas]
        else:
            return lemmas