Index

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)

B

bartz module bartz.BART module bartz.grove module bartz.jaxext module

bartz.mcmcloop module bartz.mcmcstep module bartz.prepcovars module bin_predictors() (in module bartz.prepcovars)

C

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

compute_count_trees() (in module bartz.mcmcstep) compute_likelihood_ratio() (in module bartz.mcmcstep) compute_p_prune() (in module bartz.mcmcstep) compute_partial_ratio() (in module bartz.mcmcstep) compute_prec_trees() (in module bartz.mcmcstep) count_datapoints_per_leaf() (in module bartz.mcmcstep) Counts (class in bartz.mcmcstep)

D

default_both_extractor() (in module bartz.mcmcloop)

default_onlymain_extractor() (in module bartz.mcmcloop)

E

ensure_unsigned() (in module bartz.jaxext)

evaluate_forest() (in module bartz.grove) evaluate_trace() (in module bartz.mcmcloop)

F

float_type() (in module bartz.jaxext) Forest (class in bartz.mcmcstep)

forest_fill() (in module bartz.grove) fully_used_variables() (in module bartz.mcmcstep)

G

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

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

H

huge_value() (in module bartz.jaxext)

I

init() (in module bartz.mcmcstep) is_actual_leaf() (in module bartz.grove)

is_leaves_parent() (in module bartz.grove) is_used() (in module bartz.grove)

M

make_print_callbacks() (in module bartz.mcmcloop) make_tree() (in module bartz.grove) minimal_unsigned_dtype() (in module bartz.jaxext) module bartz bartz.BART bartz.grove bartz.jaxext bartz.mcmcloop bartz.mcmcstep bartz.prepcovars

Moves (class in bartz.mcmcstep)

P

ParallelStageOut (class in bartz.mcmcstep) 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) 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)

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

scipy (class in bartz.jaxext) scipy.special (class in bartz.jaxext) scipy.stats (class in bartz.jaxext) scipy.stats.invgamma (class in bartz.jaxext) SeqStageInAllTrees (class in bartz.mcmcstep) SeqStageInPerTree (class in bartz.mcmcstep) signed_to_unsigned() (in module bartz.jaxext)

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

traverse_forest() (in module bartz.grove) traverse_tree() (in module bartz.grove)

tree_depth() (in module bartz.grove) tree_depths() (in module bartz.grove)

U

uniform_splits_from_matrix() (in module bartz.prepcovars)

unique() (in module bartz.jaxext)

V

vmap_nodoc() (in module bartz.jaxext)