from sklearn.base import BaseEstimator
def __init__(self, param1=0, param2=None):
# it is important that you name your params the same in the constructor
# stub as well as in your class variables!
self.param1 = param1
self.param2 = param2
def fit(self, data, targets=None, **kwargs):
Adjust the underlying model or method to the data.
IMPORTANT: must return self!
def predict(self, data):
Use the learned model to make predictions.
my_predictions =