Initial commit
This commit is contained in:
13
classifiers/manual.py
Normal file
13
classifiers/manual.py
Normal file
@@ -0,0 +1,13 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user