# spam_detector_ai/classifiers/base_classifier.py

from abc import ABC, abstractmethod

from joblib import dump, load


class BaseClassifier(ABC):
    VECTORIZER_PARAMS = {
        'max_features': 1500,
        'min_df': 5,
        'max_df': 0.7
    }

    def __init__(self):
        self.classifier = None
        self.vectoriser = None

    @abstractmethod
    def train(self, X_train, y_train):
        pass

    def save_model(self, model_path, vectoriser_path):
        dump(self.classifier, model_path)
        dump(self.vectoriser, vectoriser_path)

    def load_model(self, model_path, vectoriser_path):
        self.classifier = load(model_path)
        self.vectoriser = load(vectoriser_path)