SauceCat / PDPbox

python partial dependence plot toolbox
http://pdpbox.readthedocs.io/en/latest/
MIT License
840 stars 129 forks source link

PDPbox

PyPI version codecov Build Status

Python Partial Dependence Plot toolbox.

Visualize the influence of certain features on model predictions for supervised machine learning algorithms, utilizing partial dependence plots.

For a comprehensive explanation, I recommend referring to the Partial Dependence Plot (PDP) chapter in Christoph Molnar's book, Interpretable Machine Learning.

I am back! :smirk_cat:

After four years...

I'm delighted to see how popular PDPbox has become; it has exceeded all my expectations. When I first embarked on this project, it was a modest endeavor, simply to whet my appetite for real-world Python package development.

With the shift in my career path towards deep learning in 2018, I had to halt the development and maintenance of PDPbox. As I no longer actively used it and several other outstanding packages such as lime and shap were emerging.

However, as the years have passed, I have seen PDPbox gain a significant presence in the community. It's been referenced in various online courses and books, demonstrating its valuable role. Despite well-known limitations of partial dependence plots, their simplicity and intuitiveness might have made them a popular starting point for many, appealing to a broad range of audiences.

Given this, I feel a renewed sense of responsibility to revisit the project, refine the existing code, potentially add new features, and create additional tutorials. I'm excited about this next phase and look forward to contributing more to the open source community.

Installation

Reference

Testing

Test with pytest

cd <dir>/PDPbox
python -m pytest tests

Test with tox

PDPbox can be tested using tox.

Gallery