Index

_

__call__() (bartz.mcmcloop.Callback method)

A

a (bartz.mcmcstep.Forest attribute) affluence_tree (bartz.mcmcstep.Forest attribute) 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

b (bartz.mcmcstep.Forest attribute) 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) blocked_vars (bartz.mcmcstep.Forest attribute) 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) config (bartz.mcmcstep.State attribute) count_num_batches (bartz.mcmcstep.StepConfig attribute)

D

debug_gbart (class in bartz.debug) debug_mc_gbart (class in bartz.debug) depth (bartz.debug.SamplePriorX attribute)

depth_distr() (bartz.debug.debug_mc_gbart method) describe_error() (in module bartz.debug) dot_every (bartz.mcmcloop.PrintCallbackState attribute)

E

error_cov_df (bartz.mcmcstep.State attribute) error_cov_inv (bartz.mcmcstep.State attribute)

error_cov_scale (bartz.mcmcstep.State attribute) evaluate_forest() (in module bartz.grove) evaluate_trace() (in module bartz.mcmcloop)

F

forest (bartz.mcmcstep.State attribute) 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) gbart (class in bartz.BART) get_default_device() (in module bartz.jaxext)

get_device_count() (in module bartz.jaxext) grow_acc_count (bartz.mcmcstep.Forest attribute) grow_prop_count (bartz.mcmcstep.Forest attribute)

H

heap_size (bartz.debug.BARTTraceMeta attribute)

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)

K

key (bartz.debug.SamplePriorCarry attribute)

L

leaf_indices (bartz.mcmcstep.Forest attribute) leaf_prior_cov_inv (bartz.mcmcstep.Forest attribute) leaf_tree (bartz.debug.SamplePriorTrees attribute) (bartz.grove.TreeHeaps attribute) (bartz.mcmcstep.Forest attribute)

log_likelihood (bartz.mcmcstep.Forest attribute) log_s (bartz.mcmcstep.Forest attribute) log_trans_prior (bartz.mcmcstep.Forest attribute) lower (bartz.debug.SamplePriorStack attribute)

M

MainTrace (class in bartz.mcmcloop) make_default_callback() (in module bartz.mcmcloop) make_p_nonterminal() (in module bartz.mcmcstep) make_tree() (in module bartz.grove) max_split (bartz.mcmcstep.Forest attribute) mc_gbart (class in bartz.BART) mesh (bartz.mcmcstep.StepConfig attribute) min_points_per_decision_node (bartz.mcmcstep.Forest attribute) min_points_per_leaf (bartz.mcmcstep.Forest attribute) 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) (bartz.BART.mc_gbart property) (bartz.debug.BARTTraceMeta attribute) ndtri() (in module bartz.jaxext.scipy.special) next_depth (bartz.debug.SamplePriorX attribute)

node (bartz.debug.SamplePriorX attribute) nonterminal (bartz.debug.SamplePriorStack attribute) ntree (bartz.debug.BARTTraceMeta attribute) num_chains() (bartz.mcmcstep.Forest method) numcut (bartz.debug.BARTTraceMeta attribute)

O

offset (bartz.Bart attribute) (bartz.BART.mc_gbart property) (bartz.mcmcstep.State attribute)

P

p_nonterminal (bartz.mcmcstep.Forest attribute) p_propose_grow (bartz.mcmcstep.Forest attribute) 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) prec_count_num_trees (bartz.mcmcstep.StepConfig attribute) prec_num_batches (bartz.mcmcstep.StepConfig attribute) prec_scale (bartz.mcmcstep.State attribute) predict() (bartz.Bart method) (bartz.BART.mc_gbart 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) (bartz.BART.mc_gbart property) prob_test_mean (bartz.Bart property) (bartz.BART.mc_gbart property) prob_train (bartz.Bart property) (bartz.BART.mc_gbart property) prob_train_mean (bartz.Bart property) (bartz.BART.mc_gbart property) profile_mode() (in module bartz) prune_acc_count (bartz.mcmcstep.Forest attribute) prune_prop_count (bartz.mcmcstep.Forest attribute)

Q

quantilized_splits_from_matrix() (in module bartz.prepcovars)

R

report_every (bartz.mcmcloop.PrintCallbackState attribute) resid (bartz.mcmcstep.State attribute)

resid_num_batches (bartz.mcmcstep.StepConfig attribute) rho (bartz.mcmcstep.Forest attribute) 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) sigest (bartz.Bart attribute) (bartz.BART.mc_gbart property) sigma (bartz.Bart property) (bartz.BART.mc_gbart property) sigma_ (bartz.Bart property) (bartz.BART.mc_gbart property)

sigma_harmonic_mean() (bartz.debug.debug_mc_gbart method) sigma_mean (bartz.Bart property) (bartz.BART.mc_gbart property) sparse_on_at (bartz.mcmcstep.StepConfig attribute) split (bartz.debug.SamplePriorStack attribute) (class in bartz.jaxext) split_tree (bartz.debug.SamplePriorTrees attribute) (bartz.grove.TreeHeaps attribute) (bartz.mcmcstep.Forest attribute) stack (bartz.debug.SamplePriorCarry attribute) State (class in bartz.mcmcstep) step() (in module bartz.mcmcstep) StepConfig (class in bartz.mcmcstep) steps_done (bartz.mcmcstep.StepConfig attribute)

T

theta (bartz.mcmcstep.Forest attribute) 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 (bartz.debug.SamplePriorCarry attribute) 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) upper (bartz.debug.SamplePriorStack attribute)

V

var (bartz.debug.SamplePriorStack attribute) var_histogram() (in module bartz.grove) var_tree (bartz.debug.SamplePriorTrees attribute) (bartz.grove.TreeHeaps attribute) (bartz.mcmcstep.Forest attribute) varcount (bartz.Bart property) (bartz.BART.mc_gbart property)

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

X

X (bartz.mcmcstep.State attribute)

Y

y (bartz.mcmcstep.State attribute) yhat_test (bartz.Bart attribute) (bartz.BART.mc_gbart property) yhat_test_mean (bartz.Bart property) (bartz.BART.mc_gbart property)

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

Z

z (bartz.mcmcstep.State attribute)