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 bartz.testing module bin_predictors() (in module bartz.prepcovars) blocked_vars (bartz.mcmcstep.Forest attribute) burnin_trace (bartz.mcmcloop.RunMCMCResult attribute) BurninTrace (class in bartz.mcmcloop)

C

Callback (class in bartz.mcmcloop) check_trace() (in module bartz.debug) check_trees() (bartz.debug.debug_mc_gbart method)

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_distr() (bartz.debug.debug_mc_gbart method)

describe_error() (in module bartz.debug) DGP (class in bartz.testing) dot_every (bartz.mcmcloop.PrintCallbackState attribute)

E

equal_shards() (in module bartz.jaxext) 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

final_state (bartz.mcmcloop.RunMCMCResult attribute) forest (bartz.mcmcstep.State attribute) Forest (class in bartz.mcmcstep) forest_depth_distr() (in module bartz.grove) forest_fill() (in module bartz.grove)

format_tree() (in module bartz.grove) 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) gen_data() (in module bartz.testing)

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.SamplePriorTrees 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)

J

jit_active() (in module bartz.jaxext)

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)

M

main_trace (bartz.mcmcloop.RunMCMCResult attribute) MainTrace (class in bartz.mcmcloop) make_default_callback() (in module bartz.mcmcloop) make_p_nonterminal() (in module bartz.mcmcstep) 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 bartz.testing

N

ndpost (bartz.Bart property) (bartz.BART.mc_gbart property) (bartz.debug.BARTTraceMeta attribute) ndtri() (in module bartz.jaxext.scipy.special)

ntree (bartz.debug.BARTTraceMeta attribute) num_chains() (bartz.mcmcstep.Forest method) num_trees (bartz.Bart property) 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.grove) 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) 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) RunMCMCResult (class in bartz.mcmcloop)

S

sample_prior() (in module bartz.debug) SamplePriorTrees (class in bartz.debug) sigest (bartz.Bart attribute) (bartz.BART.mc_gbart property) sigma (bartz.Bart property) (bartz.BART.mc_gbart property) sigma2_mean (bartz.testing.DGP property) sigma2_pop (bartz.testing.DGP property) sigma2_pri (bartz.testing.DGP 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 (class in bartz.jaxext) split() (bartz.testing.DGP method) split_tree (bartz.debug.SamplePriorTrees attribute) (bartz.grove.TreeHeaps attribute) (bartz.mcmcstep.Forest 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.grove)

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) 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)