octopus-ml
octopus-ml copied to clipboard
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.
Octopus-ML
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
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, ...