Index

_

__call__() (bartz.mcmcloop.Callback method)

A

accept_move_and_sample_leaves() (in module bartz.mcmcstep) accept_moves_and_sample_leaves() (in module bartz.mcmcstep) accept_moves_final_stage() (in module bartz.mcmcstep) accept_moves_parallel_stage() (in module bartz.mcmcstep) accept_moves_sequential_stage() (in module bartz.mcmcstep) adapt_leaf_trees_to_grow_indices() (in module bartz.mcmcstep) ancestor_variables() (in module bartz.mcmcstep)

apply_grow_to_indices() (in module bartz.mcmcstep) apply_moves_to_leaf_indices() (in module bartz.mcmcstep) apply_moves_to_split_trees() (in module bartz.mcmcstep) autobatch() (in module bartz.jaxext) avg_acc() (bartz.debug.debug_gbart method) avg_move() (bartz.debug.debug_gbart method) avg_prop() (bartz.debug.debug_gbart method)

B

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) categorical() (in module bartz.mcmcstep) 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_gbart method) check_types() (in module bartz.debug) check_unused_node() (in module bartz.debug)

check_var_in_bounds() (in module bartz.debug) choose_leaf() (in module bartz.mcmcstep) choose_leaf_parent() (in module bartz.mcmcstep) choose_split() (in module bartz.mcmcstep) choose_variable() (in module bartz.mcmcstep) compare_resid() (bartz.debug.debug_gbart method) complete_ratio() (in module bartz.mcmcstep) compute_count_trees() (in module bartz.mcmcstep) compute_likelihood_ratio() (in module bartz.mcmcstep) compute_partial_ratio() (in module bartz.mcmcstep) compute_prec_trees() (in module bartz.mcmcstep) compute_varcount() (in module bartz.mcmcloop) count_datapoints_per_leaf() (in module bartz.mcmcstep) Counts (class in bartz.mcmcstep)

D

DataFrame (class in bartz.BART) debug_gbart (class in bartz.debug)

depth_distr() (bartz.debug.debug_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) forest_points_per_decision_node_distr() (in module bartz.debug) forest_points_per_leaf_distr() (in module bartz.debug)

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) fully_used_variables() (in module bartz.mcmcstep)

G

gammainccinv() (in module bartz.jaxext.scipy.special) gbart (class in bartz.BART)

growable_leaves() (in module bartz.mcmcstep) GrowMoves (class in bartz.mcmcstep)

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_leaves_parent() (in module bartz.grove) is_used() (in module bartz.grove)

M

MainTrace (class in bartz.mcmcloop) make_print_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

Moves (class in bartz.mcmcstep)

N

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

P

ParallelStageOut (class in bartz.mcmcstep) parse_xinfo() (in module bartz.prepcovars) points_per_decision_node_distr() (bartz.debug.debug_gbart method) (in module bartz.debug) points_per_leaf_distr() (bartz.debug.debug_gbart method) (in module bartz.debug) pop() (bartz.jaxext.split method) ppf() (bartz.jaxext.scipy.stats.invgamma static method) prec_per_leaf() (in module bartz.mcmcstep) precompute_leaf_terms() (in module bartz.mcmcstep) precompute_likelihood_terms() (in module bartz.mcmcstep) Precs (class in bartz.mcmcstep)

predict() (bartz.BART.gbart method) PreLf (class in bartz.mcmcstep) PreLk (class in bartz.mcmcstep) PreLkV (class in bartz.mcmcstep) prob_test (bartz.BART.gbart property) prob_test_mean (bartz.BART.gbart property) prob_train (bartz.BART.gbart property) prob_train_mean (bartz.BART.gbart property) propose_grow_moves() (in module bartz.mcmcstep) propose_moves() (in module bartz.mcmcstep) propose_prune_moves() (in module bartz.mcmcstep) PruneMoves (class in bartz.mcmcstep) pytree_at_set() (in module bartz.mcmcloop)

Q

quantilized_splits_from_matrix() (in module bartz.prepcovars)

R

randint_exclude() (in module bartz.mcmcstep)

randint_masked() (in module bartz.mcmcstep) 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) SeqStageInAllTrees (class in bartz.mcmcstep) SeqStageInPerTree (class in bartz.mcmcstep) Series (class in bartz.BART)

show_tree() (bartz.debug.debug_gbart method) sigma_harmonic_mean() (bartz.debug.debug_gbart method) sigma_mean (bartz.BART.gbart property) split (class in bartz.jaxext) split_range() (in module bartz.mcmcstep) State (class in bartz.mcmcstep) step() (in module bartz.mcmcstep) step_sigma() (in module bartz.mcmcstep) step_trees() (in module bartz.mcmcstep) step_z() (in module bartz.mcmcstep) sum_resid() (in module bartz.mcmcstep)

T

to_numpy() (bartz.BART.DataFrame method) (bartz.BART.Series method) Trace (class in bartz.mcmcloop) trace_depth_distr() (in module bartz.debug) trace_points_per_decision_node_distr() (in module bartz.debug) trace_points_per_leaf_distr() (in module bartz.debug) 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_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.gbart property)

varcount_mean (bartz.BART.gbart property) vmap_nodoc() (in module bartz.jaxext)

Y

yhat_test_mean (bartz.BART.gbart property)

yhat_train (bartz.BART.gbart property) yhat_train_mean (bartz.BART.gbart property)