Imbalanced-learn: A Python Toolbox to Tackle the Curse of Imbalanced Datasets in Machine Learning

Abstract

Imbalanced-learn is an open-source python toolbox aiming at providing a wide range of methods to cope with the problem of imbalanced dataset frequently encountered in machine learning and pattern recognition. The implemented state-of-the-art methods can be categorized into 4 groups: (i) under-sampling, (ii) over-sampling, (iii) combination of over- and under-sampling, and (iv) ensemble learning methods. The proposed toolbox only depends on numpy, scipy, and scikit-learn and is distributed under MIT license. Furthermore, it is fully compatible with scikit-learn and is part of the scikit-learn-contrib supported project. Documentation, unit tests as well as integration tests are provided to ease usage and contribution. The toolbox is publicly available in GitHub: https://github.com/scikit-learn-contrib/imbalanced-learn.

References in corpus (2)

• Scikit-learn: Machine Learning in Python
• SMOTE: Synthetic Minority Over-sampling Technique

Cited by in corpus (48)

• Oversampling for Imbalanced Learning Based on K-Means and SMOTE
• COVID-19 Cough Classification using Machine Learning and Global Smartphone Recordings
• Benchmarking datasets for Anomaly-based Network Intrusion Detection: KDD CUP 99 alternatives
• Self-paced Ensemble for Highly Imbalanced Massive Data Classification
• Kernel density estimation based sampling for imbalanced class distribution
• adVAE: A self-adversarial variational autoencoder with Gaussian anomaly prior knowledge for anomaly detection
• Towards Benchmark Datasets for Machine Learning Based Website Phishing Detection: An experimental study
• Discovering the Building Blocks of Atomic Systems using Machine Learning
• Audio-Based Activities of Daily Living (ADL) Recognition with Large-Scale Acoustic Embeddings from Online Videos
• Topology comparison of Twitter diffusion networks effectively reveals misleading information
• SZZ Unleashed: An Open Implementation of the SZZ Algorithm -- Featuring Example Usage in a Study of Just-in-Time Bug Prediction for the Jenkins Project
• Explainable Artificial Intelligence for Pharmacovigilance: What Features Are Important When Predicting Adverse Outcomes?
• Gamma distribution-based sampling for imbalanced data
• FIRE-DES++: Enhanced Online Pruning of Base Classifiers for Dynamic Ensemble Selection
• An Empirical Study on the Effectiveness of Data Resampling Approaches for Cross-Project Software Defect Prediction
• High-Resolution Road Vehicle Collision Prediction for the City of Montreal
• Differential diagnosis and molecular stratification of gastrointestinal stromal tumors on CT images using a radiomics approach
• PHOTONAI -- A Python API for Rapid Machine Learning Model Development
• Automatic non-invasive Cough Detection based on Accelerometer and Audio Signals
• Learning to Prescribe Interventions for Tuberculosis Patients Using Digital Adherence Data
• VFPred: A Fusion of Signal Processing and Machine Learning techniques in Detecting Ventricular Fibrillation from ECG Signals
• Deeply Supervised Skin Lesions Diagnosis with Stage and Branch Attention
• System Design for a Data-driven and Explainable Customer Sentiment Monitor
• Deep Neural Network based Cough Detection using Bed-mounted Accelerometer Measurements
• Multi-task fusion for improving mammography screening data classification
• Explainable machine learning for predicting shellfish toxicity in the Adriatic Sea using long-term monitoring data of HABs
• Road Roughness Estimation Using Machine Learning
• Tree Boosting Methods for Balanced andImbalanced Classification and their Robustness Over Time in Risk Assessment
• Heartbeat Anomaly Detection using Adversarial Oversampling
• Swarm Differential Privacy for Purpose Driven Data-Information-Knowledge-Wisdom Architecture
• Graph-Based Machine Learning Improves Just-in-Time Defect Prediction
• UnbiasedNets: A Dataset Diversification Framework for Robustness Bias Alleviation in Neural Networks
• Skew-Probabilistic Neural Networks for Learning from Imbalanced Data
• Propheticus: Machine Learning Framework for the Development of Predictive Models for Reliable and Secure Software
• Protected Attributes Tell Us Who, Behavior Tells Us How: A Comparison of Demographic and Behavioral Oversampling for Fair Student Success Modeling
• Counterfeits on Darknet Markets: A measurement between Jan-2014 and Sep-2015
• Not So Fast: A New Catalog of Meteor Persistent Trains
• B-HAR: an open-source baseline framework for in depth study of human activity recognition datasets and workflows
• Optimal 1-NN Prototypes for Pathological Geometries
• Benchmark of Data Preprocessing Methods for Imbalanced Classification
• Synthetic Information towards Maximum Posterior Ratio for deep learning on Imbalanced Data
• Reproducible Machine Learning-based Voice Pathology Detection: Introducing the Pitch Difference Feature
• Noninvasive identification of carbon-based black pigments with pump-probe microscopy
• Structural Characterization of Grain Boundaries and Machine Learning of Grain Boundary Energy and Mobility
• The Development of Spatial Attention U-Net for The Recovery of Ionospheric Measurements and The Extraction of Ionospheric Parameters
• A neural network for beam background decomposition in Belle II at SuperKEKB
• A New Constructive Heuristic driven by Machine Learning for the Traveling Salesman Problem
• Feature Selection Techniques for CR Isotope Identification with the AMS-02 Experiment in Space