import numpy as np from sklearn.base import BaseEstimator, ClassifierMixin class ManualClassifier(BaseEstimator, ClassifierMixin): def fit(self, X, y): return self def predict(self, X): return np.array([self.predict_one(msg) for msg in X]) def predict_one(self, message): return "ham"