Skip to content

Instantly share code, notes, and snippets.

@tianhuil

tianhuil/classifier_transform.py

Created November 9, 2019 05:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tianhuil/40fb1e9cea78364f60a2edbd025c370c to your computer and use it in GitHub Desktop.
Save tianhuil/40fb1e9cea78364f60a2edbd025c370c to your computer and use it in GitHub Desktop.
Download ZIP
Turns classifier's `predict_proba` into a transform
Raw
classifier_transform.py
from sklearn.base import BaseEstimator, TransformerMixin
class ClassifierTransform(BaseEstimator, TransformerMixin):
def __init__(self, clf):
self.clf = clf
def fit(self, X, y=None):
self.clf.fit(X, y)
return self
def transform(self, X):
return self.clf.predict_proba(X)
def classifier_transform(clf):
return ClassifierTransform(clf)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment