| src / bartz / BART.py |
DataFrame.to_numpy |
|
100% |
0 |
0 |
1 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
Series.to_numpy |
|
100% |
0 |
0 |
1 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.__init__ |
|
100% |
29 |
0 |
0 |
|
100% |
8 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.prob_test |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.prob_test_mean |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.prob_train |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.prob_train_mean |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.sigma |
|
100% |
9 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.sigma_ |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.sigma_mean |
|
100% |
4 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.varcount |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._compute_varcount_multichain |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._compute_varcount_multichain_flattened |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.varcount_mean |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.varprob |
|
100% |
9 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.varprob_mean |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.yhat_test_mean |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.yhat_train |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.yhat_train_mean |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart.predict |
|
100% |
6 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._process_predictor_input |
|
100% |
7 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._process_response_input |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._check_same_length |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._process_error_variance_settings |
|
69% |
29 |
9 |
0 |
|
62% |
16 |
4 |
|
67% |
| src / bartz / BART.py |
mc_gbart._check_type_settings |
|
53% |
15 |
7 |
0 |
|
60% |
10 |
4 |
|
56% |
| src / bartz / BART.py |
mc_gbart._process_sparsity_settings |
|
100% |
8 |
0 |
0 |
|
83% |
6 |
1 |
|
93% |
| src / bartz / BART.py |
mc_gbart._process_offset_settings |
|
90% |
10 |
1 |
0 |
|
83% |
6 |
1 |
|
88% |
| src / bartz / BART.py |
mc_gbart._process_leaf_sdev_settings |
|
100% |
7 |
0 |
0 |
|
83% |
6 |
1 |
|
92% |
| src / bartz / BART.py |
mc_gbart._determine_splits |
|
100% |
8 |
0 |
0 |
|
100% |
6 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._bin_predictors |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._setup_mcmc |
|
75% |
20 |
5 |
0 |
|
70% |
10 |
1 |
|
73% |
| src / bartz / BART.py |
mc_gbart._run_mcmc |
|
100% |
12 |
0 |
0 |
|
100% |
6 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._single_run_mcmc |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._add_multichain_index |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._vmapped_run_mcmc |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._vmapped_run_mcmc._partial_vmapped_run_mcmc |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._vmap_axes_for_state |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._vmap_axes_for_state.choose_vmap_index |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._predict |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._evaluate_chains_flattened |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
mc_gbart._evaluate_chains |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / BART.py |
gbart.__init__ |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / BART.py |
(no function) |
|
100% |
102 |
0 |
4 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / __init__.py |
(no function) |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
get_profile_mode |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
set_profile_mode |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
profile_mode |
|
100% |
5 |
0 |
1 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
jit_and_block_if_profiling |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
jit_and_block_if_profiling.jab_inner_wrapper |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
jit_and_block_if_profiling.jab_outer_wrapper |
|
80% |
5 |
1 |
0 |
|
75% |
4 |
1 |
|
78% |
| src / bartz / _profiler.py |
jit_if_not_profiling |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _profiler.py |
jit_if_not_profiling.wrapper |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / _profiler.py |
scan_if_not_profiling |
|
100% |
7 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / _profiler.py |
cond_if_not_profiling |
|
100% |
5 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / _profiler.py |
callback_if_not_profiling |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / _profiler.py |
(no function) |
|
100% |
20 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / _version.py |
(no function) |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
format_tree |
|
100% |
10 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
format_tree.traverse_tree |
|
79% |
33 |
7 |
0 |
|
62% |
16 |
2 |
|
73% |
| src / bartz / debug.py |
tree_actual_depth |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
forest_depth_distr |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
trace_depth_distr |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
chains_depth_distr |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
points_per_decision_node_distr |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
forest_points_per_decision_node_distr |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
forest_points_per_decision_node_distr.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
chains_points_per_decision_node_distr |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
chains_points_per_decision_node_distr.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
points_per_leaf_distr |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
forest_points_per_leaf_distr |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
forest_points_per_leaf_distr.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
chains_points_per_leaf_distr |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
chains_points_per_leaf_distr.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_types |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_sizes |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_unused_node |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_leaf_values |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_stray_nodes |
|
100% |
7 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_rule_consistency |
|
88% |
8 |
1 |
0 |
|
50% |
2 |
1 |
|
80% |
| src / bartz / debug.py |
check_rule_consistency._check_recursive |
|
100% |
10 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / debug.py |
check_num_nodes |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_var_in_bounds |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_split_in_bounds |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_tree |
|
100% |
8 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / debug.py |
describe_error |
|
0% |
1 |
1 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
check_trace |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
check_chains |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
_get_next_line |
|
75% |
4 |
1 |
0 |
|
50% |
2 |
1 |
|
67% |
| src / bartz / debug.py |
scan_BART_trees |
|
79% |
39 |
8 |
0 |
|
71% |
14 |
4 |
|
77% |
| src / bartz / debug.py |
TraceWithOffset.from_trees_trace |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
trees_BART_to_bartz |
|
100% |
26 |
0 |
0 |
|
100% |
8 |
0 |
|
100% |
| src / bartz / debug.py |
SamplePriorStack.initial |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
SamplePriorTrees.initial |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
SamplePriorCarry.initial |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
SamplePriorX.initial |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
SamplePriorX._sequence |
|
100% |
6 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / debug.py |
sample_prior_onetree |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
sample_prior_onetree.loop |
|
100% |
21 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
sample_prior_onetree.loop.write_push_stack |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
sample_prior_onetree.loop.pop_push_stack |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
sample_prior_forest |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
sample_prior |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
evaluate_forests |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
evaluate_forests._evaluate_forests |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.__init__ |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.print_tree |
|
0% |
4 |
4 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.sigma_harmonic_mean |
|
0% |
11 |
11 |
0 |
|
0% |
2 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.compare_resid |
|
100% |
9 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.avg_acc |
|
0% |
3 |
3 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.avg_acc.acc |
|
0% |
3 |
3 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.avg_prop |
|
0% |
6 |
6 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.avg_prop.prop |
|
0% |
1 |
1 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.avg_move |
|
0% |
3 |
3 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
debug_mc_gbart.depth_distr |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.points_per_decision_node_distr |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.points_per_leaf_distr |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.check_trees |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / debug.py |
debug_mc_gbart.tree_goes_bad |
|
0% |
3 |
3 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / debug.py |
(no function) |
|
100% |
107 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
make_tree |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
tree_depth |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
traverse_tree |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
traverse_tree.loop |
|
100% |
7 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
traverse_forest |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
evaluate_forest |
|
100% |
7 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / grove.py |
is_actual_leaf |
|
100% |
10 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / grove.py |
is_leaves_parent |
|
100% |
7 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
tree_depths |
|
100% |
8 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / grove.py |
is_used |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
forest_fill |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
var_histogram |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / grove.py |
(no function) |
|
100% |
23 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
vmap_nodoc |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
minimal_unsigned_dtype |
|
57% |
7 |
3 |
0 |
|
50% |
6 |
1 |
|
54% |
| src / bartz / jaxext / __init__.py |
unique |
|
100% |
9 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
unique.loop |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
split.__init__ |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
split.__len__ |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
split.pop |
|
100% |
10 |
0 |
0 |
|
100% |
6 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
_split_unpack |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
_split_shaped |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
truncated_normal_onesided |
|
100% |
12 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / __init__.py |
(no function) |
|
100% |
24 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
expand_axes |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
expand_axes.expand_axis |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
check_no_nones |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
check_no_nones.check_not_none |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
extract_size |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
extract_size.get_size |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
sum_nbytes |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
sum_nbytes.nbytes |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
next_divisor_small |
|
75% |
4 |
1 |
0 |
|
50% |
4 |
2 |
|
62% |
| src / bartz / jaxext / _autobatch.py |
next_divisor_large |
|
100% |
5 |
0 |
0 |
|
75% |
4 |
1 |
|
89% |
| src / bartz / jaxext / _autobatch.py |
next_divisor |
|
100% |
5 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
pull_nonbatched |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
pull_nonbatched.pull_nonbatched |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
push_nonbatched |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
push_nonbatched.push_nonbatched |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
move_axes_out |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
move_axes_out.move_axis_out |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
move_axes_in |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
move_axes_in.move_axis_in |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
batch |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
batch.batch |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
unbatch |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
unbatch.unbatch |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
check_same |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
check_same.check_same |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
autobatch |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
autobatch.batched_func |
|
100% |
28 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
autobatch.batched_func.loop |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / _autobatch.py |
(no function) |
|
100% |
26 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / __init__.py |
(no function) |
|
100% |
0 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
_float_type |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
_castto |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
_castto.newfunc |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
gammainccinv |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
ndtri |
|
60% |
5 |
2 |
0 |
|
50% |
2 |
1 |
|
57% |
| src / bartz / jaxext / scipy / special.py |
_ndtri |
|
100% |
27 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
_ndtri._create_polynomial |
|
100% |
4 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / jaxext / scipy / special.py |
(no function) |
|
100% |
13 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / stats.py |
invgamma.ppf |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / jaxext / scipy / stats.py |
(no function) |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
BurninTrace.from_state |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
MainTrace.from_state |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
Callback.__call__ |
|
100% |
0 |
0 |
1 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
run_mcmc |
|
100% |
12 |
0 |
0 |
|
100% |
6 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_empty_trace |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_compute_i_skip |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_run_mcmc_inner_loop |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_run_mcmc_inner_loop.loop_impl |
|
100% |
11 |
0 |
0 |
|
75% |
4 |
1 |
|
93% |
| src / bartz / mcmcloop.py |
_run_mcmc_inner_loop.loop_noop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_run_mcmc_inner_loop.loop |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_save_state_to_trace |
|
100% |
8 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_pytree_at_set |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_pytree_at_set.at_set |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
make_default_callback |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
make_default_callback.asarray_or_none |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_default_callback |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
print_callback |
|
100% |
9 |
0 |
0 |
|
100% |
6 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
print_callback.print_report |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_convert_jax_arrays_in_args |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_convert_jax_arrays_in_args.convert_jax_arrays |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_convert_jax_arrays_in_args.convert_jax_arrays.convert_jax_array |
|
60% |
5 |
2 |
0 |
|
50% |
4 |
2 |
|
56% |
| src / bartz / mcmcloop.py |
_convert_jax_arrays_in_args.new_func |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
_print_report |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
sparse_callback |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
TreesTrace.from_dataclass |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
evaluate_trace |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
evaluate_trace.loop |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
compute_varcount |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcloop.py |
(no function) |
|
100% |
51 |
0 |
2 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
init |
|
87% |
30 |
4 |
0 |
|
83% |
12 |
2 |
|
86% |
| src / bartz / mcmcstep.py |
init.make_forest |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_all_none_or_not_none |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_asarray_or_none |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_choose_suffstat_batch_size |
|
55% |
22 |
10 |
0 |
|
50% |
12 |
2 |
|
53% |
| src / bartz / mcmcstep.py |
_choose_suffstat_batch_size.get_platform |
|
78% |
9 |
2 |
0 |
|
50% |
2 |
1 |
|
73% |
| src / bartz / mcmcstep.py |
step |
|
100% |
8 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
step_trees |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
propose_moves |
|
100% |
11 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
propose_grow_moves |
|
100% |
13 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
choose_leaf |
|
100% |
9 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
growable_leaves |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
categorical |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
choose_variable |
|
100% |
6 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
fully_used_variables |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
ancestor_variables |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
ancestor_variables.loop |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
split_range |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
split_range.loop |
|
100% |
8 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
randint_exclude |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
randint_exclude.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_process_exclude |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
categorical_exclude |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
choose_split |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
compute_partial_ratio |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
propose_prune_moves |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
choose_leaf_parent |
|
100% |
11 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
randint_masked |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
accept_moves_and_sample_leaves |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
accept_moves_parallel_stage |
|
100% |
21 |
0 |
0 |
|
88% |
8 |
1 |
|
97% |
| src / bartz / mcmcstep.py |
apply_grow_to_indices |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
compute_count_trees |
|
100% |
9 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
count_datapoints_per_leaf |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_count_scan |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_count_scan.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_aggregate_scatter |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_count_vec |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_aggregate_batched_alltrees |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
compute_prec_trees |
|
100% |
9 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
prec_per_leaf |
|
67% |
3 |
1 |
0 |
|
50% |
2 |
1 |
|
60% |
| src / bartz / mcmcstep.py |
_prec_scan |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_prec_scan.loop |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_prec_vec |
|
0% |
1 |
1 |
0 |
|
100% |
0 |
0 |
|
0% |
| src / bartz / mcmcstep.py |
complete_ratio |
|
100% |
13 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
adapt_leaf_trees_to_grow_indices |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
precompute_likelihood_terms |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_chol_with_gersh |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_logdet_from_chol |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
precompute_likelihood_terms_mv |
|
100% |
10 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
precompute_likelihood_terms_mv._covariance_from_chol |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
precompute_leaf_terms |
|
100% |
6 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
precompute_leaf_terms_mv |
|
100% |
12 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
accept_moves_sequential_stage |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
accept_moves_sequential_stage.loop |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
accept_move_and_sample_leaves |
|
100% |
22 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
sum_resid |
|
100% |
4 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_aggregate_batched_onetree |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
compute_likelihood_ratio |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
compute_likelihood_ratio_mv |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
compute_likelihood_ratio_mv._quadratic_form |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
accept_moves_final_stage |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
apply_moves_to_leaf_indices |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
apply_moves_to_split_trees |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
step_sigma |
|
100% |
9 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_sample_wishart_bartlett |
|
100% |
13 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
step_z |
|
100% |
5 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
step_s |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
step_theta |
|
100% |
11 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
_log_p_lamda |
|
100% |
3 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
step_sparse |
|
100% |
5 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / mcmcstep.py |
(no function) |
|
100% |
109 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
parse_xinfo |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
quantilized_splits_from_matrix |
|
100% |
7 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / prepcovars.py |
quantilized_splits_from_matrix.quantilize |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
_quantilized_splits_from_matrix |
|
100% |
16 |
0 |
0 |
|
100% |
4 |
0 |
|
100% |
| src / bartz / prepcovars.py |
_huge_value |
|
100% |
3 |
0 |
0 |
|
100% |
2 |
0 |
|
100% |
| src / bartz / prepcovars.py |
_ensure_unsigned |
|
100% |
1 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
_signed_to_unsigned |
|
47% |
15 |
8 |
0 |
|
40% |
10 |
4 |
|
44% |
| src / bartz / prepcovars.py |
uniform_splits_from_matrix |
|
100% |
6 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
bin_predictors |
|
100% |
4 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
bin_predictors.bin_predictors |
|
100% |
2 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |
| src / bartz / prepcovars.py |
(no function) |
|
100% |
17 |
0 |
0 |
|
100% |
0 |
0 |
|
100% |