Sklearn Utilities

Supported Python versions License

Utilities for scikit-learn.

Installation

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

pip install sklearn-utilities

API

See Docs for more information.

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.
IntersectXY: drops rows where the index of X and y do not intersect. Use with feature_engine.DropMissingData.
ReindexMissingColumns: reindexes columns of X in transform() to match the columns of X in fit().
ReportNonFinite: reports non-finite values in X and/or y.
IdTransformer: a transformer that does nothing.
RecursiveFitSubtractRegressor: a regressor that recursively fits a regressor and subtracts the prediction from the target.
SmartMultioutputEstimator: a MultiOutputEstimator that supports tuple of arrays in predict() and supports pandas Series and DataFrame.
until_event(), since_event(): calculates the time since or until events (Series[bool])
ComposeVarEstimator: composes mean and std/var estimators.
DummyRegressorVar: DummyRegressor that returns 1.0 for std/var.
TransformedTargetRegressorVar: TransformedTargetRegressor with std/var support.
StandardScalerVar: StandardScaler with std/var support.
EvalSetWrapper, CatBoostProgressBarWrapper: wrapper that passes eval_set to fit() using train_test_split(), mainly for CatBoost. The latter shows progress bar (using tqdm) as well. Useful for early stopping. For LightGBM, see lightgbm-callbacks.

`sklearn_utilities.dataset`

add_missing_values(): adds missing values to a dataset.

`sklearn_utilities.torch`

PCATorch: faster PCA using PyTorch with GPU support.

`sklearn_utilities.torch.skorch`

SkorchReshaper, SkorchCNNReshaper: reshapes X and y for nn.Linear and nn.Conv1d/2d respectively. (For nn.Conv2d, uses np.sliding_window_view().)
AllowNaN: wraps a loss module and assign 0 to y and y_hat for indices where y contains NaN in forward()..

Contributors ✨

Thanks goes to these wonderful people (emoji key):

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

Metadata

Release files for sklearn-utilities 0.5.14

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for sklearn-utilities 0.5.14
File	Size	Uploaded
sklearn_utilities-0.5.14.tar.gz	29.3 kB	May 26, 2025	Details

Built distribution (wheel)

Table of built distributions (wheels) for sklearn-utilities 0.5.14
File	Interpreter	ABI	Platform	Reset
sklearn_utilities-0.5.14-py3-none-any.whl 40.2 kB May 26, 2025	Python 3	none	any	Details

Total release size: 69.5 kB

Release files / sklearn_utilities-0.5.14.tar.gz

Download URL	sklearn_utilities-0.5.14.tar.gz
Size	29.3 kB
Tags	Source
SHA-256 checksum How to use checksums	`c485d19e139e79e39229ad3ff07849b8fa855a9a4075b96a3bfab3662df763a0`
BLAKE2b-256 checksum How to use checksums	`1be40250cb39ea5a2615eaacbc535d03d78c373d492b97e449e9f32b6afe763a`
Upload date	May 26, 2025
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.1.0 CPython/3.12.9`

Release files / sklearn_utilities-0.5.14-py3-none-any.whl

Download URL	sklearn_utilities-0.5.14-py3-none-any.whl
Size	40.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`c729fe8234c4d3dab008a275f4946d1535eadb2f541bf43f0304e29b6a985662`
BLAKE2b-256 checksum How to use checksums	`14f77a5d4f51b4043275c8e8b7c069fa13716984aecc5f5479b6cff97bc3a4f0`
Upload date	May 26, 2025
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.1.0 CPython/3.12.9`