Skip to main content

Automated Machine Learning for Supervised tasks

Project description


Build Status PyPI version PyPI pyversions

Automated Machine Learning

mljar-supervised is an Automated Machine Learning python package. It can train ML models for:

  • binary classification,
  • multi-class classification,
  • regression.

What's good in it?

  • mljar-supervised creates markdown reports from AutoML training. The example of AutoML leaderboard summary:

AutoML leaderboard

The example for Decision Tree summary: Decision Tree summary

The example for LightGBM summary: Decision Tree summary

  • This package is computing Baseline for your data. So you will know if you need Machine Learning or not! You will know how good are your ML models comparing to the Baseline. The Baseline is computed based on prior class distribution for classification, and simple mean for regression.
  • This package is training simple Decision Trees with max_depth <= 5, so you can easily visualize them with amazing dtreeviz to better understand your data.
  • The mljar-supervised is using simple linear regression and include its coefficients in the summary report, so you can check which features are used the most in the linear model.
  • It is using a vast set of algorithms: Random Forest, Extra Trees, LightGBM, Xgboost, CatBoost (Neural Networks will be added soon).
  • It can do features preprocessing, like: missing values imputation and converting categoricals. What is more, it can also handle target values preprocessing (You won't believe how often it is needed!). For example, converting categorical target into numeric.
  • It can tune hyper-parameters with not-so-random-search algorithm (random-search over defined set of values) and hill climbing to fine-tune final models.
  • It can compute Ensemble based on greedy algorithm from Caruana paper.
  • It cares about explainability of models: for every algorithm, the feature importance is computed based on permutation. Additionally, for every algorithm the SHAP explanations are computed: feature importance, dependence plots, and decision plots (explanations can be switched off with explain_level parameter).

Quick example

There is a simple interface available with fit and predict methods.

import pandas as pd
from supervised.automl import AutoML

df = pd.read_csv("", skipinitialspace=True)

X = df[df.columns[:-1]]
y = df["income"]

automl = AutoML(results_path="directory_with_reports"), y)

predictions = automl.predict(X)

For details please check AutoML API Docs.



From PyPi repository:

pip install mljar-supervised

From source code:

git clone
cd mljar-supervised
python install

Installation for development

git clone
virtualenv venv --python=python3.6
source venv/bin/activate
pip install -r requirements.txt
pip install -r requirements_dev.txt


The mljar-supervised is an open-source project created by MLJAR. We care about ease of use in the Machine Learning. The provides a beautiful and simple user interface for building machine learning models.

Project details

Release history Release notifications | RSS feed

This version


Download files

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

Source Distribution

mljar-supervised-0.4.0.tar.gz (52.0 kB view hashes)

Uploaded Source

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