Jupyter/VSCode/PyCharm user? Check out my latest project to build modular data pipelines!

sklearn-evaluation

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

Supports Python 3.6 and higher.

Documentation here.

Install

pip install sklearn-evaluation

Features

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