Skip to content

Instantly share code, notes, and snippets.

@conormm
Created April 13, 2021 17:27
Show Gist options
  • Save conormm/8e7773c351deb3439e6a8b3ac0f2c96a to your computer and use it in GitHub Desktop.
Save conormm/8e7773c351deb3439e6a8b3ac0f2c96a to your computer and use it in GitHub Desktop.
from sklearn.base import BaseEstimator, ClassifierMixin, RegressorMixin
class CustomClassifier(BaseEstimator, ClassifierMixin):
def __init__(self):
pass
def fit(self, X, y=None):
pass
def predict(self, X, y=None):
pass
def predict_proba(self, X, y=None):
pass
class CustomRegressor(BaseEstimator, RegressorMixin):
def __init__(self):
pass
def fit(self, X, y=None):
pass
def predict(self, X, y=None):
pass
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment