bartz 0.11.0
Super-fast BART (Bayesian Additive Regression Trees) in Python
Navigation
BART vectoriZed
Guide
Examples
Reference
Changelog
Development
Other BART packages
Related Topics
Documentation overview
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
|
Z
_
__call__() (bartz.mcmcloop.Callback method)
(bartz.prepcovars.BinnerFactory method)
A
a (bartz.mcmcstep.Forest attribute)
A_distr (bartz.testing.Params attribute)
A_separate (bartz.testing.Params attribute)
A_shared (bartz.testing.Params attribute)
accumulator (bartz.mcmcloop.PrintCallbackState attribute)
(bartz.mcmcloop.TqdmCallbackState attribute)
affluence_tree (bartz.mcmcstep.Forest attribute)
alpha (bartz.stochtree.MeanForestParams attribute)
(bartz.testing.Gamma attribute)
auto_gpu_target (bartz.mcmcstep.PallasReduction attribute)
AutoBatchedReduction (class in bartz.mcmcstep)
AutoOneHotReduction (class in bartz.mcmcstep)
axes_from_dataclass() (bartz.grove.TreesTrace method)
B
b (bartz.mcmcstep.Forest attribute)
backend (bartz.mcmcstep.PallasReduction attribute)
bar_id (bartz.mcmcloop.TqdmCallbackState attribute)
Bart (class in bartz)
BARTModel (class in bartz.stochtree)
BARTTraceMeta (class in bartz.debug)
bartz
module
bartz.BART
module
bartz.debug
module
bartz.grove
module
bartz.mcmcloop
module
bartz.mcmcstep
module
bartz.prepcovars
module
bartz.stochtree
module
bartz.testing
module
BatchedReduction (class in bartz.mcmcstep)
batches_inner (bartz.mcmcstep.AutoBatchedReduction attribute)
(bartz.mcmcstep.BatchedReduction attribute)
beta (bartz.stochtree.MeanForestParams attribute)
beta_distr (bartz.testing.Params attribute)
beta_separate (bartz.testing.Params attribute)
beta_shared (bartz.testing.Params attribute)
beta_sm (bartz.mcmcstep.AutoBatchedReduction attribute)
bin() (bartz.prepcovars.Binner method)
(bartz.prepcovars.GivenSplitsBinner method)
(bartz.prepcovars.RangeEvenBinner method)
(bartz.prepcovars.UniqueQuantileBinner method)
binary (bartz.mcmcstep.OutcomeType attribute)
binary_indices (bartz.mcmcstep.State attribute)
binary_y (bartz.mcmcstep.State attribute)
Binner (class in bartz.prepcovars)
BinnerFactory (class in bartz.prepcovars)
block_size (bartz.mcmcstep.PallasReduction attribute)
blocked_vars (bartz.mcmcstep.Forest attribute)
burnin_trace (bartz.mcmcloop.RunMCMCResult attribute)
BurninTrace (class in bartz.mcmcloop)
C
Callback (class in bartz.mcmcloop)
CallbackState (in module bartz.mcmcloop)
check_trace() (in module bartz.grove)
columns (bartz.DataFrame property)
compute_varcount() (in module bartz.mcmcloop)
config (bartz.mcmcstep.State attribute)
Constant (class in bartz.testing)
contiguous (bartz.mcmcstep.AutoBatchedReduction attribute)
(bartz.mcmcstep.BatchedReduction attribute)
continuous (bartz.mcmcstep.OutcomeType attribute)
count (bartz.mcmcloop.StatsAccumulator attribute)
count_reduction_config (bartz.mcmcstep.StepConfig attribute)
count_tree (bartz.mcmcstep.Forest attribute)
D
data_sharded (bartz.mcmcstep.StepConfig property)
DataFrame (class in bartz)
describe_error() (in module bartz.grove)
DGP (class in bartz.testing)
DiagWishart (class in bartz.mcmcstep)
DiscreteUniform (class in bartz.testing)
Distr (class in bartz.testing)
dot_every (bartz.mcmcloop.PrintCallbackState attribute)
dump() (bartz.Bart method)
E
error_chol (bartz.testing.Params attribute)
error_cov_inv (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.State attribute)
error_distr (bartz.testing.Params attribute)
error_scale (bartz.testing.DGP attribute)
EvaluableTrace (class in bartz.mcmcloop)
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_mean_leaves() (in module bartz.grove)
format_tree() (in module bartz.grove)
fourth_moment (bartz.testing.Constant property)
(bartz.testing.Gamma property)
(bartz.testing.ScaleDistr property)
(bartz.testing.SpikeSlab property)
from_dataclass() (bartz.grove.TreesTrace class method)
from_peff() (bartz.testing.Constant class method)
(bartz.testing.Gamma class method)
(bartz.testing.ScaleDistr class method)
(bartz.testing.SpikeSlab class method)
from_standard_normal() (bartz.testing.DiscreteUniform method)
(bartz.testing.Distr method)
(bartz.testing.Normal method)
(bartz.testing.Uniform method)
from_state() (bartz.mcmcloop.BurninTrace class method)
(bartz.mcmcloop.MainTrace class method)
from_trees_trace() (bartz.debug.TraceWithOffset class method)
G
gamma (bartz.mcmcstep.AutoBatchedReduction attribute)
Gamma (class in bartz.testing)
gamma_distr (bartz.testing.Params attribute)
gamma_separate (bartz.testing.Params attribute)
gamma_shared (bartz.testing.Params attribute)
gbart (class in bartz.BART)
gen_data() (in module bartz.testing)
gen_data_from_params() (in module bartz.testing)
gen_params() (in module bartz.testing)
GeneralParams (class in bartz.stochtree)
get_error_sdev() (bartz.Bart method)
get_latent_prec() (bartz.Bart method)
GivenSplitsBinner (class in bartz.prepcovars)
global_var_samples (bartz.stochtree.BARTModel attribute)
grow_acc_count (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
grow_prop (bartz.mcmcloop.StatsReport attribute)
grow_prop_count (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
H
has_chains (bartz.debug.TraceWithOffset attribute)
(bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest property)
(bartz.mcmcstep.State property)
has_rfx (bartz.stochtree.BARTModel attribute)
heap_size (bartz.debug.BARTTraceMeta attribute)
het_shape (bartz.testing.Params attribute)
het_strength (bartz.testing.Params attribute)
I
include_mean_forest (bartz.stochtree.BARTModel attribute)
include_variance_forest (bartz.stochtree.BARTModel attribute)
init() (in module bartz.mcmcstep)
initial() (bartz.mcmcloop.StatsAccumulator class method)
inv_sdev_scale (bartz.mcmcstep.State attribute)
is_actual_leaf() (in module bartz.grove)
is_leaves_parent() (in module bartz.grove)
is_multivariate() (in module bartz.grove)
is_sampled() (bartz.stochtree.BARTModel method)
K
keep_every (bartz.stochtree.GeneralParams attribute)
kurtosis (bartz.testing.DiscreteUniform property)
(bartz.testing.Distr property)
(bartz.testing.Normal property)
(bartz.testing.Uniform property)
L
lambda_ (bartz.testing.Params attribute)
latent_samples (bartz.PredictKind attribute)
leaf_indices (bartz.mcmcstep.Forest attribute)
leaf_prior_cov_inv (bartz.mcmcstep.Forest attribute)
leaf_tree (bartz.grove.TreeHeaps attribute)
(bartz.grove.TreesTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
link (bartz.stochtree.OutcomeModel attribute)
load() (bartz.Bart class method)
log_likelihood (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
log_s (bartz.mcmcstep.Forest attribute)
log_trans_prior (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
M
m (bartz.testing.DiscreteUniform attribute)
main_trace (bartz.mcmcloop.RunMCMCResult attribute)
MainTrace (class in bartz.mcmcloop)
make_p_nonterminal() (in module bartz.mcmcstep)
make_print_callback() (in module bartz.mcmcloop)
make_tqdm_callback() (in module bartz.mcmcloop)
max_depth (bartz.stochtree.MeanForestParams attribute)
max_leaves (bartz.mcmcloop.StatsReport attribute)
max_split (bartz.mcmcstep.Forest attribute)
(bartz.prepcovars.Binner attribute)
(bartz.prepcovars.GivenSplitsBinner attribute)
(bartz.prepcovars.RangeEvenBinner attribute)
(bartz.prepcovars.UniqueQuantileBinner attribute)
(bartz.testing.QuantizedData attribute)
mc_gbart (class in bartz.BART)
mean (bartz.PredictKind attribute)
mean_leaves (bartz.mcmcloop.StatsReport attribute)
mean_samples (bartz.PredictKind attribute)
MeanForestParams (class in bartz.stochtree)
mesh (bartz.debug.TraceWithOffset attribute)
(bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.StepConfig attribute)
method (bartz.mcmcstep.OneHotReduction attribute)
min_batch_size (bartz.mcmcstep.AutoBatchedReduction attribute)
min_matmul_bins (bartz.mcmcstep.AutoOneHotReduction attribute)
min_points_per_decision_node (bartz.mcmcstep.Forest attribute)
min_points_per_leaf (bartz.mcmcstep.Forest attribute)
min_samples_leaf (bartz.stochtree.MeanForestParams attribute)
module
bartz
bartz.BART
bartz.debug
bartz.grove
bartz.mcmcloop
bartz.mcmcstep
bartz.prepcovars
bartz.stochtree
bartz.testing
move_acc (bartz.mcmcloop.StatsReport attribute)
mu (bartz.testing.DGP attribute)
mulin (bartz.testing.DGP attribute)
mulin_separate (bartz.testing.DGP attribute)
mulin_shared (bartz.testing.DGP attribute)
muquad (bartz.testing.DGP attribute)
muquad_separate (bartz.testing.DGP attribute)
muquad_shared (bartz.testing.DGP attribute)
N
n_inner (bartz.mcmcstep.OneHotReduction attribute)
n_samples (bartz.mcmcloop.StatsReport attribute)
n_save (bartz.Bart property)
name (bartz.Series property)
ndpost (bartz.Bart property)
(bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
(bartz.debug.BARTTraceMeta attribute)
Normal (class in bartz.testing)
NotSampledError
ntree (bartz.debug.BARTTraceMeta attribute)
nu (bartz.mcmcstep.DiagWishart attribute)
(bartz.mcmcstep.Wishart attribute)
num_batches (bartz.mcmcstep.BatchedReduction attribute)
num_blocks (bartz.mcmcstep.PallasReduction attribute)
num_burnin (bartz.stochtree.BARTModel attribute)
num_chains (bartz.Bart property)
(bartz.mcmcloop.StatsReport attribute)
(bartz.stochtree.BARTModel attribute)
(bartz.stochtree.GeneralParams attribute)
num_chains() (bartz.mcmcstep.State method)
num_gfr (bartz.stochtree.BARTModel attribute)
num_mcmc (bartz.stochtree.BARTModel attribute)
num_samples (bartz.stochtree.BARTModel attribute)
num_trees (bartz.Bart property)
(bartz.stochtree.MeanForestParams attribute)
numcut (bartz.debug.BARTTraceMeta attribute)
O
offset (bartz.Bart property)
(bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.State attribute)
(bartz.testing.Params attribute)
OneHotReduction (class in bartz.mcmcstep)
outcome (bartz.stochtree.OutcomeModel attribute)
outcome_model (bartz.stochtree.BARTModel attribute)
(bartz.stochtree.GeneralParams attribute)
outcome_samples (bartz.PredictKind attribute)
outcome_type (bartz.testing.Params attribute)
OutcomeModel (class in bartz.stochtree)
OutcomeType (class in bartz.mcmcstep)
P
p (bartz.mcmcloop.StatsReport attribute)
p_nonterminal (bartz.mcmcstep.Forest attribute)
p_propose_grow (bartz.mcmcstep.Forest attribute)
PallasReduction (class in bartz.mcmcstep)
params (bartz.testing.DGP attribute)
Params (class in bartz.testing)
partition (bartz.testing.Params attribute)
peff (bartz.mcmcloop.StatsReport attribute)
pi (bartz.testing.SpikeSlab attribute)
points_per_node_distr() (in module bartz.grove)
ppf() (bartz.testing.DiscreteUniform method)
(bartz.testing.Distr method)
(bartz.testing.Normal method)
(bartz.testing.Uniform method)
prec_count_num_trees (bartz.mcmcstep.StepConfig attribute)
prec_reduction_config (bartz.mcmcstep.StepConfig attribute)
prec_scale (bartz.mcmcstep.State attribute)
prec_tree (bartz.mcmcstep.Forest attribute)
predict() (bartz.Bart method)
(bartz.BART.gbart method)
(bartz.BART.mc_gbart method)
(bartz.stochtree.BARTModel method)
PredictKind (class in bartz)
print_callback() (in module bartz.mcmcloop)
PrintCallbackState (class in bartz.mcmcloop)
prob_test (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
prob_test_mean (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
prob_train (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
prob_train_mean (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
probit_outcome_model (bartz.stochtree.BARTModel attribute)
prune_acc_count (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
prune_prop_count (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
Q
q (bartz.testing.Params attribute)
quantize() (bartz.testing.DGP method)
(bartz.testing.DiscreteUniform method)
(bartz.testing.Distr method)
(bartz.testing.Normal method)
(bartz.testing.Uniform method)
QuantizedData (class in bartz.testing)
R
random_seed (bartz.stochtree.GeneralParams attribute)
RangeEvenBinner (class in bartz.prepcovars)
rate (bartz.mcmcstep.DiagWishart attribute)
(bartz.mcmcstep.Wishart attribute)
ReductionConfig (class in bartz.mcmcstep)
report() (bartz.mcmcloop.StatsAccumulator method)
report_every (bartz.mcmcloop.PrintCallbackState attribute)
(bartz.mcmcloop.TqdmCallbackState attribute)
reset_if() (bartz.mcmcloop.StatsAccumulator method)
resid (bartz.mcmcstep.State attribute)
resid_reduction_config (bartz.mcmcstep.StepConfig attribute)
rho (bartz.mcmcstep.Forest attribute)
run_mcmc() (in module bartz.mcmcloop)
RunMCMCResult (class in bartz.mcmcloop)
S
s (bartz.testing.Params attribute)
s_distr (bartz.testing.Params attribute)
sample() (bartz.stochtree.BARTModel method)
(bartz.testing.Constant method)
(bartz.testing.DiscreteUniform method)
(bartz.testing.Distr method)
(bartz.testing.Gamma method)
(bartz.testing.Normal method)
(bartz.testing.ScaleDistr method)
(bartz.testing.SpikeSlab method)
(bartz.testing.Uniform method)
sample_prior() (in module bartz.debug)
sample_sigma2_global (bartz.stochtree.BARTModel attribute)
sample_sigma2_leaf (bartz.stochtree.MeanForestParams attribute)
sampled (bartz.stochtree.BARTModel attribute)
ScaleDistr (class in bartz.testing)
sequential_unroll (bartz.mcmcstep.StepConfig attribute)
Series (class in bartz)
sigest (bartz.BART.gbart attribute)
(bartz.BART.mc_gbart attribute)
sigma (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
sigma2_eps (bartz.testing.Params attribute)
sigma2_global_scale (bartz.stochtree.GeneralParams attribute)
sigma2_global_shape (bartz.stochtree.GeneralParams attribute)
sigma2_init (bartz.stochtree.BARTModel attribute)
(bartz.stochtree.GeneralParams attribute)
sigma2_leaf_init (bartz.stochtree.MeanForestParams attribute)
sigma2_lin (bartz.testing.Params attribute)
sigma2_mean (bartz.testing.Params attribute)
sigma2_pop (bartz.testing.Params attribute)
sigma2_pri (bartz.testing.Params attribute)
sigma2_quad (bartz.testing.Params attribute)
sigma_ (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
sigma_mean (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
sparse_on_at (bartz.mcmcstep.StepConfig attribute)
SpikeSlab (class in bartz.testing)
split() (bartz.testing.DGP method)
split_tree (bartz.grove.TreeHeaps attribute)
(bartz.grove.TreesTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
standardize (bartz.stochtree.BARTModel attribute)
(bartz.stochtree.GeneralParams attribute)
State (class in bartz.mcmcstep)
StatsAccumulator (class in bartz.mcmcloop)
StatsReport (class in bartz.mcmcloop)
step() (in module bartz.mcmcstep)
StepConfig (class in bartz.mcmcstep)
steps_done (bartz.mcmcstep.StepConfig attribute)
sums (bartz.mcmcloop.StatsAccumulator attribute)
T
theta (bartz.mcmcloop.BurninTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
to_numpy() (bartz.DataFrame method)
(bartz.Series method)
tqdm_callback() (in module bartz.mcmcloop)
TqdmCallbackState (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)
TreeHeaps (class in bartz.grove)
trees_BART_to_bartz() (in module bartz.debug)
TreesTrace (class in bartz.grove)
U
Uniform (class in bartz.testing)
UniqueQuantileBinner (class in bartz.prepcovars)
update() (bartz.mcmcloop.StatsAccumulator method)
update_every (bartz.mcmcloop.TqdmCallbackState attribute)
V
value (bartz.mcmcstep.DiagWishart attribute)
(bartz.mcmcstep.Wishart attribute)
var_histogram() (in module bartz.grove)
var_tree (bartz.grove.TreeHeaps attribute)
(bartz.grove.TreesTrace attribute)
(bartz.mcmcloop.MainTrace attribute)
(bartz.mcmcstep.Forest attribute)
var_v (bartz.testing.Params attribute)
varcount (bartz.Bart property)
(bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
varcount_mean (bartz.Bart property)
(bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
variable_weights (bartz.stochtree.GeneralParams attribute)
varprob (bartz.Bart property)
(bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
(bartz.mcmcloop.MainTrace attribute)
varprob_mean (bartz.Bart property)
(bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
W
Wishart (class in bartz.mcmcstep)
X
X (bartz.mcmcstep.State attribute)
x (bartz.testing.DGP attribute)
(bartz.testing.QuantizedData attribute)
x_distr (bartz.testing.Params attribute)
Y
y (bartz.testing.DGP attribute)
(bartz.testing.QuantizedData attribute)
y_bar (bartz.stochtree.BARTModel attribute)
y_hat_test (bartz.stochtree.BARTModel attribute)
y_hat_train (bartz.stochtree.BARTModel attribute)
y_std (bartz.stochtree.BARTModel attribute)
yhat_test (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
yhat_test_mean (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
yhat_train (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
yhat_train_mean (bartz.BART.gbart property)
(bartz.BART.mc_gbart property)
Z
z (bartz.mcmcstep.State attribute)
(bartz.testing.DGP attribute)