Skip to content

feat: add scikit-learn compatibility (#34)#103

Open
SahilYadav-op wants to merge 1 commit into
yoavram-lab:mainfrom
SahilYadav-op:fix/sklearn-compatibility
Open

feat: add scikit-learn compatibility (#34)#103
SahilYadav-op wants to merge 1 commit into
yoavram-lab:mainfrom
SahilYadav-op:fix/sklearn-compatibility

Conversation

@SahilYadav-op
Copy link
Copy Markdown

Fixes #34

Summary:
This PR enables full integration with the scikit-learn ecosystem by ensuring core estimators and preprocessing utilities inherit from BaseEstimator and TransformerMixin.

Changes:

  • Added BaseEstimator and TransformerMixin inheritance to VBPCA, AutoEncoder, and all MissingAware scalers/encoders.
  • Implemented custom get_params and set_params for VBPCA to support its variable keyword options.
  • Added a new test suite tests/test_sklearn_compat.py covering clone() and Pipeline round-trips.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add get_params / set_params for sklearn Pipeline compatibility

1 participant