Skip to content

Tool for whitebox (binning + logreg) model development

License

Notifications You must be signed in to change notification settings

sberbank-ai-lab/AutoMLWhitebox

Repository files navigation

Sberbank version of AutoWoE

GitHub all releases PyPI - Downloads

This is the repository for AutoWoE library, developed by Sber AI Lab AutoML group. This library can be used for automatic creation of interpretable ML model based on feature binning, WoE features transformation, feature selection and Logistic Regression.

Authors: Vakhrushev Anton, Grigorii Penkin, Alexander Kirilin

Library setup can be done by one of three scenarios below:

  1. Installation from PyPI:
pip install autowoe
  1. Installation from source code

First of all you need to install git and poetry.

# Load LAMA source code
git clone https://github.com/sberbank-ai-lab/AutoMLWhitebox.git

cd AutoMLWhiteBox/

# !!!Choose only one item!!!

# 1. Recommended: Create virtual environment inside your project directory
poetry config virtualenvs.in-project true

# 2. Global installation: Don't create virtual environment
poetry config virtualenvs.create false --local

# For more information read poetry docs

# Install WhiteBox
poetry install

Usage tutorials are in Jupyter notebooks in the repository root. For parameters description take a look at parameters_info.md.

Bugs / Questions / Suggestions::