Skip to main content

Helpers to speed up and structure machine learning projects

Project description

ML Helper

Helpers to speed up and structure machine learning projects.

The library is available in Pypi


The easiest way to install ml-helper is through pip

pip install ml-helper

To use it in your project, you must first import the library

from ml_helper.helper import Helper

And then create a Helper object with a dictionary of keys related to your project

KEYS = {
    'SEED': 1,
    'TARGET': 'y',
    'METRIC': 'r2',
    'TIMESERIES': True,
    'SPLITS': 5

hp = Helper(KEYS)

After this, you may use the helper object's many functions


ML-Helper requires:

  • Python (>3.5)
  • Numpy (>=1.16)
  • Pandas (>=0.23.4)
  • Seaborn (>=0.9)
  • Scikit-learn (>=0.20)
  • Natplotlib (>=3)
  • Scipy (>=1)
  • Imblearn
  • Vecstack


The functionality is separated into 4 groups:

  • Data Exploration
    • Missing Data
    • Boxplot of numerical variables
    • Coefficient of variation
    • Correlation (numerical and categorical)
    • Under Represented Features
    • Target Variable Distribution
    • Feature Importance
    • PCA Component Variance
  • Data Preparation
    • Convert features to categories
    • Drop multiple columns
  • Modeling
    • Cross Validation (with stratified kfolds, or time series split depending on use case)
      • Randomized Grid Search
    • Pipeline: Collection of models and pipeline steps that get performed and scored
    • Predict: Predict on unseen data
    • Stack Predict: Build a stacked model and perform a prediction
  • Regression
    • Plots for predictions
  • Classification
    • ROC Curve
    • Classification Report
  • Others
    • Select features based on types
    • Split X and y
    • Plot models/pipelines

Working Examples

If you wish to see the library in use, you may view the notebooks in the examples section.

Also, you can see the implementation in their corresponding Kaggle Kernels:

ML-Helper Coding Style

Ml-Helper complies to PEP8 and uses black for coding standards


SemVer is used for versioning.


This project is licensed under the MIT License - see the License file for details

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ml-helper-0.0.23.tar.gz (8.5 kB view hashes)

Uploaded Source

Built Distribution

ml_helper-0.0.23-py3-none-any.whl (8.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page