nullnull/simstring

View on GitHub
simstring/feature_extractor/character_ngram.py

Summary

Maintainability
A
0 mins
Test Coverage
from .base import BaseFeatureExtractor

SENTINAL_CHAR = " "  # non breaking space

class CharacterNgramFeatureExtractor(BaseFeatureExtractor):
    def __init__(self, n=2):
        self.n = n

    def features(self, string):
        return self._each_cons(SENTINAL_CHAR + string + SENTINAL_CHAR, self.n)