Interpretable machine learning models (imodels) 🔍

Straightforward implementations of interpretable ML models + demos of how to use various interpretability techniques. Code is optimized for readability. Pull requests welcome!

Implementations of imodels • Demo notebooks • Docs

Implementations of interpretable models

Scikit-learn style wrappers/implementations of different interpretable models. Docs available here. The interpretable models within the imodels folder can be easily installed and used:

pip install git+https://github.com/csinva/interpretability-implementations-demos

from imodels import RuleListClassifier, RuleFit, GreedyRuleList, SkopeRules, SLIM, IRFClassifier
model = RuleListClassifier()  # initialize Bayesian Rule List
model.fit(X_train, y_train)   # fit model
preds = model.predict(X_test) # discrete predictions: shape is (n_test, 1)
preds_proba = model.predict_proba(X_test) # predicted probabilities: shape is (n_test, n_classes)

• bayesian rule list (based on this implementation) - learn a compact rule list
• rulefit (based on this implementation) - find rules from a decision tree and build a linear model with them
• skope-rules (based on this implementation) - extracts rules from base estimators (e.g. decision trees) then tries to deduplicate them
• sparse integer linear model (simple implementation with cvxpy)
• greedy rule list (based on this implementation) - uses CART to learn a list (only a single path), rather than a decision tree
• (in progress) optimal classification tree (based on this implementation) - learns succinct trees using global optimization rather than greedy heuristics
• iterative random forest (based on this implementation)
• see readmes in individual folders within imodels for details

Demo notebooks

The demos are contained in 3 main notebooks, following this cheat-sheet:

1. model_based.ipynb - how to use different interpretable models and examples with the imodels package
2. posthoc.ipynb - different simple analyses to interpret a trained model
3. uncertainty.ipynb - code to get uncertainty estimates for a model

References / further reading

• high-level review on interpretable machine learning
• book on interpretable machine learning
• review on black-blox explanation methods
• review on variable importance
• for updates, star the repo, see this related repo, or follow @chandan_singh96