https://github.com/ploomber/sklearn-evaluation

Machine learning model evaluation made easy: plots, tables, HTML reports, experiment tracking and Jupyter notebook analysis.
https://github.com/ploomber/sklearn-evaluation

data-science deep-learning jupyter-notebook machine-learning pytorch scikit-learn sklearn tensorflow

Last synced: 26 days ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/ploomber/sklearn-evaluation
• Owner: ploomber
• License: apache-2.0
• Created: 2015-09-04T16:33:42.000Z (about 9 years ago)
• Default Branch: master
• Last Pushed: 2024-02-28T13:59:10.000Z (8 months ago)
• Last Synced: 2024-04-14T03:08:55.882Z (7 months ago)
• Topics: data-science, deep-learning, jupyter-notebook, machine-learning, pytorch, scikit-learn, sklearn, tensorflow
• Language: Python
• Homepage: https://sklearn-evaluation.ploomber.io
• Size: 20.1 MB
• Stars: 447
• Watchers: 15
• Forks: 54
• Open Issues: 4
• Metadata Files:
  • Readme: README.md
  • Changelog: CHANGELOG.md
  • Contributing: CONTRIBUTING.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-python-machine-learning-resources - GitHub - 20% open · ⏱️ 22.08.2022): (模型的可解释性)

README

        
# sklearn-evaluation

![CI](https://github.com/ploomber/sklearn-evaluation/workflows/CI/badge.svg)

[![Documentation Status](https://readthedocs.org/projects/sklearn-evaluation/badge/?version=latest)](https://sklearn-evaluation.readthedocs.io/en/latest/?badge=latest)

[![PyPI version](https://badge.fury.io/py/sklearn-evaluation.svg)](https://badge.fury.io/py/sklearn-evaluation)

[![Coverage Status](https://coveralls.io/repos/github/ploomber/sklearn-evaluation/badge.svg)](https://coveralls.io/github/ploomber/sklearn-evaluation)

[![Twitter](https://img.shields.io/twitter/follow/edublancas?label=Follow&style=social)](https://twitter.com/intent/user?screen_name=ploomber)

[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)

[![Downloads](https://static.pepy.tech/badge/sklearn-evaluation)](https://pepy.tech/project/sklearn-evaluation)

> [!TIP]

> Deploy AI apps for free on [Ploomber Cloud!](https://ploomber.io/?utm_medium=github&utm_source=sklearn-evaluation)



  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.



    



![confusion matrix](examples/cm.png)

# Install  

```bash

pip install sklearn-evaluation

```

# Features

* [Plotting](https://sklearn-evaluation.ploomber.io/en/latest/classification/basic.html) (confusion matrix, feature importances, precision-recall, roc, elbow curve, silhouette plot)

* Report generation ([example](https://htmlpreview.github.io/?https://github.com/ploomber/sklearn-evaluation/blob/master/examples/report.html))

* [Evaluate grid search results](https://sklearn-evaluation.ploomber.io/en/latest/classification/optimization.html)

* [Track experiments using a local SQLite database](https://sklearn-evaluation.ploomber.io/en/latest/comparison/SQLiteTracker.html)

* [Analyze notebooks output](https://sklearn-evaluation.ploomber.io/en/latest/comparison/NotebookCollection.html)

* [Query notebooks with SQL](https://sklearn-evaluation.ploomber.io/en/latest/comparison/nbdb.html)