Index

_

__call__() (bartz.mcmcloop.Callback method)

A

autobatch() (in module bartz.jaxext) avg_acc() (bartz.debug.debug_mc_gbart method)

avg_move() (bartz.debug.debug_mc_gbart method) avg_prop() (bartz.debug.debug_mc_gbart method)

B

Bart (class in bartz) BARTTraceMeta (class in bartz.debug) bartz module bartz.BART module bartz.debug module bartz.grove module bartz.jaxext module

bartz.jaxext.scipy.special module bartz.jaxext.scipy.stats module bartz.mcmcloop module bartz.mcmcstep module bartz.prepcovars module bin_predictors() (in module bartz.prepcovars) BurninTrace (class in bartz.mcmcloop)

C

Callback (class in bartz.mcmcloop) check() (in module bartz.debug) check_leaf_values() (in module bartz.debug) check_num_nodes() (in module bartz.debug) check_rule_consistency() (in module bartz.debug) check_sizes() (in module bartz.debug) check_split_in_bounds() (in module bartz.debug) check_stray_nodes() (in module bartz.debug)

check_trace() (in module bartz.debug) check_tree() (in module bartz.debug) check_trees() (bartz.debug.debug_mc_gbart method) check_types() (in module bartz.debug) check_unused_node() (in module bartz.debug) check_var_in_bounds() (in module bartz.debug) compare_resid() (bartz.debug.debug_mc_gbart method) compute_varcount() (in module bartz.mcmcloop)

D

debug_gbart (class in bartz.debug) debug_mc_gbart (class in bartz.debug)

depth_distr() (bartz.debug.debug_mc_gbart method) describe_error() (in module bartz.debug)

E

evaluate_forest() (in module bartz.grove)

evaluate_trace() (in module bartz.mcmcloop)

F

Forest (class in bartz.mcmcstep) forest_depth_distr() (in module bartz.debug) forest_fill() (in module bartz.grove) format_tree() (in module bartz.debug)

from_dataclass() (bartz.mcmcloop.TreesTrace class method) from_state() (bartz.mcmcloop.BurninTrace class method) (bartz.mcmcloop.MainTrace class method) from_trees_trace() (bartz.debug.TraceWithOffset class method)

G

gammainccinv() (in module bartz.jaxext.scipy.special)

get_default_device() (in module bartz.jaxext) get_device_count() (in module bartz.jaxext)

I

init() (in module bartz.mcmcstep) initial() (bartz.debug.SamplePriorCarry class method) (bartz.debug.SamplePriorStack class method) (bartz.debug.SamplePriorTrees class method) (bartz.debug.SamplePriorX class method)

invgamma (class in bartz.jaxext.scipy.stats) is_actual_leaf() (in module bartz.grove) is_key() (in module bartz.jaxext) is_leaves_parent() (in module bartz.grove) is_used() (in module bartz.grove)

M

MainTrace (class in bartz.mcmcloop) make_default_callback() (in module bartz.mcmcloop) make_tree() (in module bartz.grove) minimal_unsigned_dtype() (in module bartz.jaxext) module bartz bartz.BART bartz.debug bartz.grove bartz.jaxext bartz.jaxext.scipy.special bartz.jaxext.scipy.stats bartz.mcmcloop bartz.mcmcstep bartz.prepcovars

N

ndpost (bartz.Bart property)

ndtri() (in module bartz.jaxext.scipy.special) num_chains() (bartz.mcmcstep.Forest method)

P

parse_xinfo() (in module bartz.prepcovars) points_per_decision_node_distr() (bartz.debug.debug_mc_gbart method) points_per_leaf_distr() (bartz.debug.debug_mc_gbart method) points_per_node_distr() (in module bartz.debug) pop() (bartz.jaxext.split method) ppf() (bartz.jaxext.scipy.stats.invgamma static method) predict() (bartz.Bart method)

print_callback() (in module bartz.mcmcloop) print_tree() (bartz.debug.debug_mc_gbart method) PrintCallbackState (class in bartz.mcmcloop) prob_test (bartz.Bart property) prob_test_mean (bartz.Bart property) prob_train (bartz.Bart property) prob_train_mean (bartz.Bart property) profile_mode() (in module bartz)

Q

quantilized_splits_from_matrix() (in module bartz.prepcovars)

R

run_mcmc() (in module bartz.mcmcloop)

S

sample_prior() (in module bartz.debug) sample_prior_forest() (in module bartz.debug) sample_prior_onetree() (in module bartz.debug) SamplePriorCarry (class in bartz.debug) SamplePriorStack (class in bartz.debug) SamplePriorTrees (class in bartz.debug) SamplePriorX (class in bartz.debug) scan_BART_trees() (in module bartz.debug)

sigma (bartz.Bart property) sigma_ (bartz.Bart property) sigma_harmonic_mean() (bartz.debug.debug_mc_gbart method) sigma_mean (bartz.Bart property) split (class in bartz.jaxext) State (class in bartz.mcmcstep) step() (in module bartz.mcmcstep) StepConfig (class in bartz.mcmcstep)

T

Trace (class in bartz.mcmcloop) TraceWithOffset (class in bartz.debug) traverse_forest() (in module bartz.grove) traverse_tree() (in module bartz.grove) tree_actual_depth() (in module bartz.debug) tree_depth() (in module bartz.grove)

tree_depths() (in module bartz.grove) tree_goes_bad() (bartz.debug.debug_mc_gbart method) TreeHeaps (class in bartz.grove) trees_BART_to_bartz() (in module bartz.debug) TreesTrace (class in bartz.mcmcloop) truncated_normal_onesided() (in module bartz.jaxext)

U

uniform_splits_from_matrix() (in module bartz.prepcovars)

unique() (in module bartz.jaxext)

V

var_histogram() (in module bartz.grove) varcount (bartz.Bart property) varcount_mean (bartz.Bart property)

varprob (bartz.Bart property) varprob_mean (bartz.Bart property) vmap_nodoc() (in module bartz.jaxext)

Y

yhat_test_mean (bartz.Bart property)

yhat_train (bartz.Bart property) yhat_train_mean (bartz.Bart property)