Skip to main content

Utilities for scikit-learn.

Project description

Sklearn Utilities

CI Status Documentation Status Test coverage percentage

Poetry black pre-commit

PyPI Version Supported Python versions License

Utilities for scikit-learn.

Installation

Install this via pip (or your favourite package manager):

pip install sklearn-utilities

API

  • EstimatorWrapperBase: base class for wrappers. Redirects all attributes which are not in the wrapper to the wrapped estimator.
  • DataFrameWrapper: tries to convert every estimator output to a pandas DataFrame or Series.
  • FeatureUnionPandas: a FeatureUnion that works with pandas DataFrames.
  • IncludedColumnTransformerPandas, ExcludedColumnTransformerPandas: select columns by name.
  • AppendPredictionToX: appends the prediction of y to X.
  • AppendXPredictionToX: appends the prediction of X to X.
  • DropByNoisePrediction: drops columns which has high importance in predicting noise.
  • DropMissingColumns: drops columns with missing values above a threshold.
  • DropMissingRowsY: drops rows with missing values in y. Use feature_engine.DropMissingData for X.
  • IdTransformer: a transformer that does nothing.
  • RecursiveFitSubtractRegressor: a regressor that recursively fits a regressor and subtracts the prediction from the target.
  • until_event(), since_event(): calculates the time since or until events (Series[bool])

Contributors ✨

Thanks goes to these wonderful people (emoji key):

This project follows the all-contributors specification. Contributions of any kind welcome!

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sklearn_utilities-0.1.0.tar.gz (13.9 kB view hashes)

Uploaded Source

Built Distribution

sklearn_utilities-0.1.0-py3-none-any.whl (17.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page