bartz 802d6dd
Super-fast BART (Bayesian Additive Regression Trees) in Python
Navigation
BART vectoriZed
Guide
Reference
Changelog
Development
Other BART packages
Related Topics
Documentation overview
Quick search
Index
A
|
B
|
C
|
E
|
F
|
G
|
H
|
I
|
L
|
M
|
P
|
Q
|
R
|
S
|
T
|
U
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)
count_datapoints_per_leaf() (in module bartz.mcmcstep)
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)
fully_used_variables() (in module bartz.mcmcstep)
G
gbart (class in bartz.BART)
grow_move() (in module bartz.mcmcstep)
growable_leaves() (in module 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)
L
LeafDict (class in bartz.jaxext)
M
make_simple_print_callback() (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
P
precompute_leaf_terms() (in module bartz.mcmcstep)
precompute_likelihood_terms() (in module bartz.mcmcstep)
predict() (bartz.BART.gbart method)
prune_move() (in module bartz.mcmcstep)
pure_callback_ufunc() (in module bartz.jaxext)
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_moves() (in module bartz.mcmcstep)
sample_sigma() (in module bartz.mcmcstep)
sample_trees() (in module bartz.mcmcstep)
signed_to_unsigned() (in module bartz.jaxext)
split_range() (in module bartz.mcmcstep)
step() (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)