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

abess

Fast Best-Subset Selection Library

skrebate

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

evalml

EvalML is an AutoML library written in python.

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.

arfs

All Relevant Feature Selection

divik

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

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.

shap-hypetune

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

sklearn-genetic-opt

ML hyperparameters tuning and features selection, using evolutionary algorithms.

fastcan

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

dsr-feature-eng-ml

Machine learning model evaluation and feature engineering framework with hyperparameter tuning, data balancing, and feature importance analysis.

stg

feature selection using stochastic gates

ball

Statistical Inference and Sure Independence Screening via Ball Statistics

kts

Interactive ML Toolset

autofeat

Linear Prediction Model with Automated Feature Engineering and Selection Capabilities

adapt

Awesome Domain Adaptation Python Toolbox

kxy

A toolkit to boost the productivity of machine learning engineers.

boostaroota

A fast xgboost feature selection algorithm

lofo-importance

Leave One Feature Out Importance

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.