Source code for OpenAttack.metric.selectors.edit_distance

from .base import MetricSelector
from ...text_process.tokenizer import get_default_tokenizer

class EditDistance(MetricSelector):
    """
    :English: `Levenshtein` ( :py:class:`.PunctTokenizer` )
    :Chinese: `Levenshtein` ( :py:class:`.JiebaTokenizer` )
    """

    def _select(self, lang):
        from ..algorithms.levenshtein import Levenshtein
        return Levenshtein( get_default_tokenizer(lang) )