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.

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

mrmr-selection

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

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.

divik

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

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

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.

evalml

EvalML is an AutoML library written in python.

shap-hypetune

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

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.

fastcan

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

arfs

All Relevant Feature Selection

boostaroota

A fast xgboost feature selection algorithm

aaanalysis

Python framework for interpretable protein prediction

sklearn-genetic-opt

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

kxy

A toolkit to boost the productivity of machine learning engineers.

kts

Interactive ML Toolset

autofeat

Linear Prediction Model with Automated Feature Engineering and Selection Capabilities

ball

Statistical Inference and Sure Independence Screening via Ball Statistics

skfeature-chappers

Feature selection with scikit-learn

adapt

Awesome Domain Adaptation Python Toolbox