Add new vi plots (#196)

* add new vi plots * fix tests * add missing file
pymc-devs · Nov 25, 2024 · 4ef2dd0 · 4ef2dd0
1 parent 282b2ef
commit 4ef2dd0
Show file tree

Hide file tree

Showing 3 changed files with 285 additions and 101 deletions.
diff --git a/pymc_bart/__init__.py b/pymc_bart/__init__.py
@@ -17,11 +17,14 @@
 from pymc_bart.pgbart import PGBART
 from pymc_bart.split_rules import ContinuousSplitRule, OneHotSplitRule, SubsetSplitRule
 from pymc_bart.utils import (
+    compute_variable_importance,
     plot_convergence,
     plot_dependence,
     plot_ice,
     plot_pdp,
+    plot_scatter_submodels,
     plot_variable_importance,
+    plot_variable_inclusion,
 )
 
 __all__ = [
@@ -30,11 +33,14 @@
     "ContinuousSplitRule",
     "OneHotSplitRule",
     "SubsetSplitRule",
+    "compute_variable_importance",
     "plot_convergence",
     "plot_dependence",
     "plot_ice",
     "plot_pdp",
+    "plot_scatter_submodels",
     "plot_variable_importance",
+    "plot_variable_inclusion",
 ]
 __version__ = "0.7.1"