Skip to content

Instantly share code, notes, and snippets.

@data-hound
Last active January 22, 2021 17:53
Scikeras Tutorial-2: DataTransformers Template
from sklearn.base import BaseEstimator, TransformerMixin
from scikeras.wrappers import KerasClassifier
### Multi-Output Classifier
class MultiOutputTransformer(BaseEstimator, TransformerMixin):
#define your transformer
class MultiOutputClassifier(KerasClassifier):
@property
def target_encoder(self):
return MultiOutputTransformer(...)
### Multi-Input Classifier
class MultiInputTransformer(BaseEstimator, TransformerMixin):
#define your transformer
class MultiInputClassifier(KerasClassifier):
@property
def feature_encoder(self):
return MultiInputTransformer(...)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment