Skip to content

gershonc/octopus-ml

Repository files navigation

Octopus-ML

PyPI Latest Release License Python Version Code style: black Binder Downloads Colab Demo

Set of handy ML and data tools - starting from data exploration, visualization, pre-processing, hyper parameter tuning, modeling and all the way to final ML model evaluation

Check out the octopus-ml demo notebook on Colab google colab logo

Installation

The module can be easily installed with pip:

> pip install octopus-ml

This module depends on Scikit-learn, NumPy, Pandas, TQDM, lightGBM as defualt classifier. Optionally you can get also some nice visualisations if you have Seaborn installed.

Usage

The module contains ML and Data related methods:

from octopus_ml import plot_imp, adjusted_classes, cv, cv_plot, roc_curve_plot, ...



Selected visualizations: