# Factory for making new Classifier objects
from classifiers.standard.bert_classifier import BertGeneralClassifier
from classifiers.standard.bert_random_classifier import BertRandomClassifier
from classifiers.standard.fast_text_classifier import FastTextClassifier
from classifiers.standard.bert_mean_pool_classifier import BertMeanPoolClassifier
from classifiers.standard.bert_concat_classifier import BertConcatClassifier
from classifiers.sequential.bert_sequence_classifier import BertSequenceClassifier


class ClassifierFactory:
    def __init__(self, config):
        self._classifiers = {"bert-general": BertGeneralClassifier,
                             "bert-random": BertRandomClassifier,
                             "fast-text": FastTextClassifier,
                             "bert-mean-pool": BertMeanPoolClassifier,
                             "bert-concat": BertConcatClassifier,
                             "bert-seq": BertSequenceClassifier}
        self.config = config

    def register_classifier(self, name, classifier):
        self._classifiers[name] = classifier

    def make_classifier(self, name):
        classifier = self._classifiers.get(name)
        if not classifier:
            raise ValueError(format)
        return classifier(self.config)