sklearn-evaluation

[!TIP] Deploy AI apps for free on Ploomber Cloud!

Join our community | Newsletter | Contact us | Docs | Blog | Website | YouTube

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking, and Jupyter notebook analysis.

Supports Python 3.7 and higher. Tested on Linux, macOS and Windows.

Note: Recent versions likely work on Python 3.6; however, 0.8.2 was the latest version tested with such Python version.

Install

pip install sklearn-evaluation

Features

• Plotting (confusion matrix, feature importances, precision-recall, roc, elbow curve, silhouette plot)
• Report generation (example)
• Evaluate grid search results
• Track experiments using a local SQLite database
• Analyze notebooks output
• Query notebooks with SQL