10000 Add new vi plots (#196) · pymc-devs/pymc-bart@4ef2dd0 · GitHub
[go: up one dir, main page]
Skip to content

Commit 4ef2dd0

Browse files
aloctavodiaaloctavodia
authored
Add new vi plots (#196)
* add new vi plots * fix tests * add missing file
1 parent 282b2ef commit 4ef2dd0

File tree

3 files changed

+285
-101
lines changed

3 files changed

+285
-101
lines changed

‎pymc_bart/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,14 @@
1717
from pymc_bart.pgbart import PGBART
1818
from pymc_bart.split_rules import ContinuousSplitRule, OneHotSplitRule, SubsetSplitRule
1919
from pymc_bart.utils import (
20+
compute_variable_importance,
2021
plot_convergence,
2122
plot_dependence,
2223
plot_ice,
2324
plot_pdp,
25+
plot_scatter_submodels,
2426
plot_variable_importance,
27+
plot_variable_inclusion,
2528
)
2629

2730
__all__ = [
@@ -30,11 +33,14 @@
3033
"ContinuousSplitRule",
3134
"OneHotSplitRule",
3235
"SubsetSplitRule",
36+
"compute_variable_importance",
3337
"plot_convergence",
3438
"plot_dependence",
3539
"plot_ice",
3640
"plot_pdp",
41+
"plot_scatter_submodels",
3742
"plot_variable_importance",
43+
"plot_variable_inclusion",
3844
]
3945
__version__ = "0.7.1"
4046

0 commit comments

Comments
 (0)
0