/scikit-learn-extensions

A collection of tools that extend the scikit-learn functionality.

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

scikit-learn-extensions

Extensions of scikit-learn. It includes a class to simplify the search of model space as well as various oversamplers and evaluation metrics.

Travis

Prerequisites

  • Pandas
  • Scikit-Learn
  • Imbalanced-Learn

Installation

git clone https://github.com/gdouzas/scikit-learn-extensions
cd scikit-learn-extensions
python3 -m venv pyenv
source pyenv/bin/activate
pip install .