Skip to content

Instantly share code, notes, and snippets.

@mrgloom

mrgloom/kmtransformer.py

Forked from larsmans/kmtransformer.py
Last active August 31, 2015 13:46
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 mrgloom/d0d4a4b62ad559b0caf1 to your computer and use it in GitHub Desktop.
Save mrgloom/d0d4a4b62ad559b0caf1 to your computer and use it in GitHub Desktop.
Download ZIP
k-means feature mapper for scikit-learn
Raw
kmtransformer.py
from sklearn.base import BaseEstimator, TransformerMixin
from sklearn.metrics.pairwise import rbf_kernel
class KMeansTransformer(BaseEstimator, TransformerMixin):
def __init__(self, centroids):
self.centroids = centroids
def fit(self, X, y=None):
return self
def transform(self, X, y=None):
return rbf_kernel(X, self.centroids)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment