Feature Selection Python Packages

feature-engine

Feature engineering and selection open-source Python library compatible with sklearn.

powershap

A power-full Shapley feature selection method.

mrmr-selection

mRMR (minimum-Redundancy-Maximum-Relevance) for automatic feature selection at scale.

upgini

Data search & enrichment library for Machine Learning → Easily find and add relevant features to your ML & AI pipeline from hundreds of public and premium external data sources, including open & commercial LLMs

evalml

EvalML is an AutoML library written in python.

abess

Fast Best-Subset Selection Library

fastcan

A fast canonical-correlation-based search algorithm for feature selection, system identification, data pruning, etc.

skrebate

A scikit-learn-compatible Python implementation of ReBATE, a suite of Relief-based feature selection algorithms for Machine Learning.

shap-hypetune

A python package for simultaneous Hyperparameters Tuning and Features Selection for Gradient Boosting Models.

divik

Divisive Intelligent K-Means algorithm (DiviK) for joint feature selection and clustering of heavily multidimensional data.

arfs

All Relevant Feature Selection

featurewiz

Use advanced feature engineering strategies and select best features from your data set with a single line of code. Created by Ram Seshadri. Collaborators welcome.

sklearn-genetic-opt

Hyperparameter optimization and feature selection for scikit-learn using evolutionary algorithms. A modern alternative to GridSearchCV and RandomizedSearchCV.

nvtabular

NVTabular is a feature engineering and preprocessing library for tabular data designed to quickly and easily manipulate terabyte scale datasets used to train deep learning based recommender systems.

kxy

A toolkit to boost the productivity of machine learning engineers.

desbordante

Desbordante is a high-performance data profiler that is capable of discovering many different patterns in data using various algorithms. It also allows to run data cleaning scenarios using these algorithms. Desbordante has a console version and an easy-to-use web application.

pyopls

A Python 3 implementation of orthogonal projection to latent structures

kts

Interactive ML Toolset

nlptoolkit-classification

Machine learning library for classification tasks

boostaroota

A fast xgboost feature selection algorithm

sk-transformers

A collection of pandas & scikit-learn compatible transformers for preprocessing and feature engineering 🛠

ball

Statistical Inference and Sure Independence Screening via Ball Statistics

skfeature-chappers

Feature selection with scikit-learn

adapt

Awesome Domain Adaptation Python Toolbox