14 lines
325 B
Python
14 lines
325 B
Python
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"
|