Skip to content

Instantly share code, notes, and snippets.

@data-hound
Last active January 22, 2021 17:53
Show Gist options
  • Save data-hound/18c29b28deec11b0559faf7f6bca18cb to your computer and use it in GitHub Desktop.
Save data-hound/18c29b28deec11b0559faf7f6bca18cb to your computer and use it in GitHub Desktop.
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