Skip to content

A collection of handy ML and data visualization and validation tools. Go ahead and train, evaluate and validate your ML models and data with minimal effort.

License

Notifications You must be signed in to change notification settings

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: