recognizers/base.py

"""
Source: https://github.com/ZurichNLP/recognizing-semantic-differences
MIT License
Copyright (c) 2023 University of Zurich
"""

from typing import List

from tqdm import tqdm

from recognizers.utils import DifferenceSample


class DifferenceRecognizer:

    def __str__(self):
        raise NotImplemented

    def predict(self,
                a: str,
                b: str,
                **kwargs,
                ) -> DifferenceSample:
        raise NotImplemented

    def predict_all(self,
                    a: List[str],
                    b: List[str],
                    **kwargs,
                    ) -> List[DifferenceSample]:
        assert len(a) == len(b)
        predictions = []
        for i in tqdm(list(range(len(a)))):
            prediction = self.predict(a[i], b[i], **kwargs)
            predictions.append(prediction)
        return predictions