h0rton
latest
Installation
How to train
h0rton package
Feedback
API Reference
h0rton
Docs
»
Index
Edit on GitHub
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
_
__call__() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLCPU method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.DiagonalGaussianNLL method)
(h0rton.losses.DiagonalGaussianNLLCPU method)
(h0rton.losses.DiagonalGaussianNLLNative method)
(h0rton.losses.DoubleGaussianNLL method)
(h0rton.losses.DoubleGaussianNLLCPU method)
(h0rton.losses.DoubleGaussianNLLNative method)
(h0rton.losses.DoubleLowRankGaussianNLL method)
(h0rton.losses.DoubleLowRankGaussianNLLNative method)
(h0rton.losses.FullRankGaussianNLL method)
(h0rton.losses.FullRankGaussianNLLCPU method)
(h0rton.losses.FullRankGaussianNLLNative method)
(h0rton.losses.LowRankGaussianNLL method)
(h0rton.losses.LowRankGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll.DiagonalGaussianNLL method)
(h0rton.losses.gaussian_nll.DoubleGaussianNLL method)
(h0rton.losses.gaussian_nll.DoubleLowRankGaussianNLL method)
(h0rton.losses.gaussian_nll.FullRankGaussianNLL method)
(h0rton.losses.gaussian_nll.LowRankGaussianNLL method)
(h0rton.losses.gaussian_nll_cpu.BaseGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_cpu.DiagonalGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_cpu.DoubleGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_cpu.FullRankGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.DiagonalGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.DoubleGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.DoubleLowRankGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.FullRankGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.LowRankGaussianNLLNative method)
__enter__() (h0rton.script_utils.HiddenPrints method)
__exit__() (h0rton.script_utils.HiddenPrints method)
__getitem__() (h0rton.trainval_data.tdlmc_data.TDLMCData method)
(h0rton.trainval_data.TDLMCData method)
(h0rton.trainval_data.XYData method)
(h0rton.trainval_data.xy_data.XYData method)
__len__() (h0rton.trainval_data.tdlmc_data.TDLMCData method)
(h0rton.trainval_data.TDLMCData method)
(h0rton.trainval_data.XYData method)
(h0rton.trainval_data.xy_data.XYData method)
_reorder_measured_td_to_tdlmc() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
A
all_teams (in module corner_metric_simplified)
args (in module corner_metric_simplified)
asinh() (in module h0rton.trainval_data.data_utils)
axes (in module corner_metric_simplified)
axis_lab (in module corner_metric_simplified)
B
BaseGaussianBNNPosterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.gaussian_bnn_posterior)
BaseGaussianBNNPosteriorCPU (class in h0rton.h0_inference.gaussian_bnn_posterior_cpu)
BaseGaussianNLL (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll)
BaseGaussianNLLCPU (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_cpu)
BaseGaussianNLLNative (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_native)
C
calculate_offset_from_true_image_positions() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
catlog (in module corner_metric_simplified)
chuck_images() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
color_list (in module corner_metric_simplified)
combine_lenses() (in module h0rton.h0_inference.h0_utils)
convert_to_dataframe() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_parser)
,
[1]
corner_metric_simplified (module)
CosmoConverter (class in h0rton.h0_inference.h0_utils)
couple_lens_mass_light_centroids() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
D
DiagonalGaussianBNNPosterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.gaussian_bnn_posterior)
DiagonalGaussianBNNPosteriorCPU (class in h0rton.h0_inference.gaussian_bnn_posterior_cpu)
DiagonalGaussianNLL (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll)
DiagonalGaussianNLLCPU (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_cpu)
DiagonalGaussianNLLNative (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_native)
dict_to_array() (in module h0rton.h0_inference.mcmc_utils)
dirpath (in module corner_metric_simplified)
DoubleGaussianBNNPosterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.gaussian_bnn_posterior)
DoubleGaussianBNNPosteriorCPU (class in h0rton.h0_inference.gaussian_bnn_posterior_cpu)
DoubleGaussianNLL (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll)
DoubleGaussianNLLCPU (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_cpu)
DoubleGaussianNLLNative (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_native)
DoubleLowRankGaussianBNNPosterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.gaussian_bnn_posterior)
DoubleLowRankGaussianBNNPosteriorCPU (class in h0rton.h0_inference.gaussian_bnn_posterior_cpu)
DoubleLowRankGaussianNLL (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll)
DoubleLowRankGaussianNLLNative (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_native)
E
evaluate() (h0rton.h0_inference.mcmc_utils.HybridBNNPenalty method)
F
fig (in module corner_metric_simplified)
folders (in module corner_metric_simplified)
format_lens_model() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
format_results_for_tdlmc_metrics() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_parser)
,
[1]
format_submission() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_metrics)
ForwardModelingPosterior (class in h0rton.h0_inference.forward_modeling_posterior)
from_dict() (h0rton.h0_inference.h0_posterior.H0Posterior class method)
(h0rton.h0_inference.H0Posterior class method)
from_file() (h0rton.configs.test_config.TestConfig class method)
(h0rton.configs.TestConfig class method)
(h0rton.configs.TrainValConfig class method)
(h0rton.configs.train_val_config.TrainValConfig class method)
FullRankGaussianBNNPosterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.gaussian_bnn_posterior)
FullRankGaussianBNNPosteriorCPU (class in h0rton.h0_inference.gaussian_bnn_posterior_cpu)
FullRankGaussianNLL (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll)
FullRankGaussianNLLCPU (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_cpu)
FullRankGaussianNLLNative (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_native)
G
gaussian() (in module h0rton.h0_inference)
(in module h0rton.h0_inference.plotting_utils)
gaussian_ll_pdf() (in module h0rton.h0_inference.h0_utils)
get_1d_mapping_fig() (in module h0rton.train_utils)
(in module h0rton.train_utils.logging_utils)
get_accuracy() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_metrics)
get_baobab_config() (in module h0rton.infer_h0_precision_ceiling)
get_cosmo_mcmc_kwargs() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
get_D_dt() (h0rton.h0_inference.h0_utils.CosmoConverter method)
get_goodness() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_metrics)
get_H0() (h0rton.h0_inference.h0_utils.CosmoConverter method)
get_h0_sample() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
get_h0_sample_truth() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
get_hpd_interval() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.DiagonalGaussianBNNPosterior method)
(h0rton.h0_inference.DoubleGaussianBNNPosterior method)
(h0rton.h0_inference.DoubleLowRankGaussianBNNPosterior method)
(h0rton.h0_inference.FullRankGaussianBNNPosterior method)
(h0rton.h0_inference.LowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DiagonalGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DoubleGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DoubleLowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.FullRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.LowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DiagonalGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DoubleGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DoubleLowRankGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.FullRankGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.LowRankGaussianBNNPosteriorCPU method)
get_idx_for_params() (in module h0rton.h0_inference.mcmc_utils)
get_img_pos() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
get_lens_kwargs() (in module h0rton.h0_inference.mcmc_utils)
get_lens_light_mcmc_kwargs() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
get_lens_mass_mcmc_kwargs() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
get_light_kwargs() (in module h0rton.h0_inference.mcmc_utils)
get_logdet() (in module h0rton.train_utils)
(in module h0rton.train_utils.logging_utils)
get_lognormal_stats() (in module h0rton.h0_inference.h0_utils)
get_lognormal_stats_naive() (in module h0rton.h0_inference.h0_utils)
get_mae() (in module h0rton.train_utils)
(in module h0rton.train_utils.logging_utils)
get_normal_stats() (in module h0rton.h0_inference.h0_utils)
get_normal_stats_naive() (in module h0rton.h0_inference.h0_utils)
get_precision() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_metrics)
get_ps_kwargs() (in module h0rton.h0_inference.mcmc_utils)
get_ps_kwargs_src_plane() (in module h0rton.h0_inference.mcmc_utils)
get_ps_mcmc_kwargs() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
get_source_mcmc_kwargs() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
get_special_kwargs() (in module h0rton.h0_inference.mcmc_utils)
gridshape (in module corner_metric_simplified)
H
h0_rung1 (in module corner_metric_simplified)
H0Posterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.h0_posterior)
h0rton (module)
h0rton.combine_lenses (module)
h0rton.configs (module)
h0rton.configs.test_config (module)
h0rton.configs.train_val_config (module)
h0rton.h0_inference (module)
h0rton.h0_inference.forward_modeling_posterior (module)
h0rton.h0_inference.gaussian_bnn_posterior (module)
h0rton.h0_inference.gaussian_bnn_posterior_cpu (module)
h0rton.h0_inference.h0_posterior (module)
h0rton.h0_inference.h0_utils (module)
h0rton.h0_inference.mcmc_utils (module)
h0rton.h0_inference.plotting_utils (module)
h0rton.infer_h0_forward_modeling (module)
h0rton.infer_h0_mcmc_default (module)
h0rton.infer_h0_precision_ceiling (module)
h0rton.infer_h0_simple_mc_truth (module)
h0rton.infer_h0_tdlmc (module)
h0rton.losses (module)
h0rton.losses.gaussian_nll (module)
h0rton.losses.gaussian_nll_cpu (module)
h0rton.losses.gaussian_nll_native (module)
h0rton.models (module)
h0rton.models.bayesian_resnet (module)
h0rton.script_utils (module)
h0rton.summarize (module)
h0rton.tdlmc_utils (module)
h0rton.tdlmc_utils.reorder_images (module)
,
[1]
h0rton.tdlmc_utils.tdlmc_metrics (module)
h0rton.tdlmc_utils.tdlmc_parser (module)
,
[1]
h0rton.tests (module)
h0rton.tests.test_trainval_data (module)
h0rton.tests.test_trainval_data.test_data_utils (module)
h0rton.tests.test_trainval_data.test_xy_data (module)
h0rton.train (module)
h0rton.train_utils (module)
h0rton.train_utils.checkpoint_utils (module)
h0rton.train_utils.logging_utils (module)
h0rton.trainval_data (module)
h0rton.trainval_data.data_utils (module)
h0rton.trainval_data.tdlmc_data (module)
h0rton.trainval_data.xy_data (module)
HiddenPrints (class in h0rton.script_utils)
HybridBNNPenalty (class in h0rton.h0_inference.mcmc_utils)
I
input_dir (in module corner_metric_simplified)
interpret_pred() (in module h0rton.train_utils)
(in module h0rton.train_utils.logging_utils)
K
kwargs_lens_init (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior attribute)
kwargs_lens_light_init (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior attribute)
kwargs_ps_init (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior attribute)
kwargs_source_init (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior attribute)
kwargs_special_init (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior attribute)
L
label (in module corner_metric_simplified)
label_name (in module corner_metric_simplified)
load_state_dict() (in module h0rton.train_utils)
(in module h0rton.train_utils.checkpoint_utils)
load_state_dict_test() (in module h0rton.train_utils)
(in module h0rton.train_utils.checkpoint_utils)
log_parameterize_Y_cols() (in module h0rton.trainval_data.data_utils)
logsigmoid() (in module h0rton.h0_inference.gaussian_bnn_posterior_cpu)
LowRankGaussianBNNPosterior (class in h0rton.h0_inference)
(class in h0rton.h0_inference.gaussian_bnn_posterior)
LowRankGaussianBNNPosteriorCPU (class in h0rton.h0_inference.gaussian_bnn_posterior_cpu)
LowRankGaussianNLL (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll)
LowRankGaussianNLLNative (class in h0rton.losses)
(class in h0rton.losses.gaussian_nll_native)
M
ma (in module corner_metric_simplified)
main() (in module h0rton.combine_lenses)
(in module h0rton.infer_h0_forward_modeling)
(in module h0rton.infer_h0_mcmc_default)
(in module h0rton.infer_h0_precision_ceiling)
(in module h0rton.infer_h0_simple_mc_truth)
(in module h0rton.infer_h0_tdlmc)
(in module h0rton.summarize)
(in module h0rton.train)
metric (in module corner_metric_simplified)
metric_target (in module corner_metric_simplified)
N
n (in module corner_metric_simplified)
nll_diagonal() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLCPU method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll_cpu.BaseGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
nll_full_rank() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLCPU method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll_cpu.BaseGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
nll_low_rank() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
nll_mixture() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLCPU method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll_cpu.BaseGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
nll_mixture_low_rank() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
num_boxes (in module corner_metric_simplified)
num_plots (in module corner_metric_simplified)
P
parse_args() (in module h0rton.combine_lenses)
(in module h0rton.infer_h0_tdlmc)
(in module h0rton.summarize)
(in module h0rton.train)
parse_closed_box() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_parser)
,
[1]
parse_inference_args() (in module h0rton.script_utils)
parse_open_box() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_parser)
,
[1]
parser (in module corner_metric_simplified)
plot_D_dt_histogram() (in module h0rton.h0_inference)
(in module h0rton.h0_inference.plotting_utils)
plot_forward_modeling_comparisons() (in module h0rton.h0_inference)
(in module h0rton.h0_inference.plotting_utils)
plot_h0_histogram() (in module h0rton.h0_inference)
(in module h0rton.h0_inference.plotting_utils)
plot_mcmc_corner() (in module h0rton.h0_inference)
(in module h0rton.h0_inference.plotting_utils)
plot_weighted_h0_histogram() (in module h0rton.h0_inference)
(in module h0rton.h0_inference.plotting_utils)
plus_1_log() (in module h0rton.trainval_data.data_utils)
posterior_name (h0rton.losses.DiagonalGaussianNLL attribute)
(h0rton.losses.DiagonalGaussianNLLCPU attribute)
(h0rton.losses.DiagonalGaussianNLLNative attribute)
(h0rton.losses.DoubleGaussianNLL attribute)
(h0rton.losses.DoubleGaussianNLLCPU attribute)
(h0rton.losses.DoubleGaussianNLLNative attribute)
(h0rton.losses.DoubleLowRankGaussianNLL attribute)
(h0rton.losses.DoubleLowRankGaussianNLLNative attribute)
(h0rton.losses.FullRankGaussianNLL attribute)
(h0rton.losses.FullRankGaussianNLLCPU attribute)
(h0rton.losses.FullRankGaussianNLLNative attribute)
(h0rton.losses.LowRankGaussianNLL attribute)
(h0rton.losses.LowRankGaussianNLLNative attribute)
(h0rton.losses.gaussian_nll.DiagonalGaussianNLL attribute)
(h0rton.losses.gaussian_nll.DoubleGaussianNLL attribute)
(h0rton.losses.gaussian_nll.DoubleLowRankGaussianNLL attribute)
(h0rton.losses.gaussian_nll.FullRankGaussianNLL attribute)
(h0rton.losses.gaussian_nll.LowRankGaussianNLL attribute)
(h0rton.losses.gaussian_nll_cpu.DiagonalGaussianNLLCPU attribute)
(h0rton.losses.gaussian_nll_cpu.DoubleGaussianNLLCPU attribute)
(h0rton.losses.gaussian_nll_cpu.FullRankGaussianNLLCPU attribute)
(h0rton.losses.gaussian_nll_native.DiagonalGaussianNLLNative attribute)
(h0rton.losses.gaussian_nll_native.DoubleGaussianNLLNative attribute)
(h0rton.losses.gaussian_nll_native.DoubleLowRankGaussianNLLNative attribute)
(h0rton.losses.gaussian_nll_native.FullRankGaussianNLLNative attribute)
(h0rton.losses.gaussian_nll_native.LowRankGaussianNLLNative attribute)
postprocess_mcmc_chain() (in module h0rton.h0_inference.mcmc_utils)
pred_to_natural_gaussian() (in module h0rton.h0_inference.h0_utils)
preset_default() (h0rton.configs.train_val_config.TrainValConfig method)
(h0rton.configs.TrainValConfig method)
R
read_from_csv() (in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.tdlmc_parser)
,
[1]
remove_outliers_from_lognormal() (in module h0rton.h0_inference.h0_utils)
remove_parameters_from_pred() (in module h0rton.h0_inference.mcmc_utils)
reorder_to_tdlmc() (in module h0rton.h0_inference.h0_utils)
(in module h0rton.tdlmc_utils)
(in module h0rton.tdlmc_utils.reorder_images)
,
[1]
required_params (h0rton.h0_inference.h0_posterior.H0Posterior attribute)
(h0rton.h0_inference.H0Posterior attribute)
rescale_01() (in module h0rton.trainval_data.data_utils)
resnet101() (in module h0rton.models)
(in module h0rton.models.bayesian_resnet)
resnet34() (in module h0rton.models)
(in module h0rton.models.bayesian_resnet)
resnet44() (in module h0rton.models)
(in module h0rton.models.bayesian_resnet)
resnet50() (in module h0rton.models)
(in module h0rton.models.bayesian_resnet)
resnet56() (in module h0rton.models)
(in module h0rton.models.bayesian_resnet)
run_mcmc() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
S
sample() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.DiagonalGaussianBNNPosterior method)
(h0rton.h0_inference.DoubleGaussianBNNPosterior method)
(h0rton.h0_inference.DoubleLowRankGaussianBNNPosterior method)
(h0rton.h0_inference.FullRankGaussianBNNPosterior method)
(h0rton.h0_inference.LowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DiagonalGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DoubleGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DoubleLowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.FullRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.LowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DiagonalGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DoubleGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DoubleLowRankGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.FullRankGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.LowRankGaussianBNNPosteriorCPU method)
sample_aniso_param() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
sample_full_rank() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
sample_H0() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
sample_kappa_ext_original() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
sample_kappa_ext_transformed() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
sample_low_rank() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
save_state_dict() (in module h0rton.train_utils)
(in module h0rton.train_utils.checkpoint_utils)
seed_everything() (in module h0rton.script_utils)
seed_name (in module corner_metric_simplified)
seed_samples() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
set_bnn_post_params() (h0rton.h0_inference.mcmc_utils.HybridBNNPenalty method)
set_cosmology_observables() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
set_kwargs_data_joint() (h0rton.h0_inference.forward_modeling_posterior.ForwardModelingPosterior method)
set_monitoring_cfg() (h0rton.configs.train_val_config.TrainValConfig method)
(h0rton.configs.TrainValConfig method)
set_sliced_pred() (h0rton.h0_inference.DiagonalGaussianBNNPosterior method)
(h0rton.h0_inference.DoubleGaussianBNNPosterior method)
(h0rton.h0_inference.DoubleLowRankGaussianBNNPosterior method)
(h0rton.h0_inference.FullRankGaussianBNNPosterior method)
(h0rton.h0_inference.LowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DiagonalGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DoubleGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.DoubleLowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.FullRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.LowRankGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DiagonalGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DoubleGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.DoubleLowRankGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.FullRankGaussianBNNPosteriorCPU method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.LowRankGaussianBNNPosteriorCPU method)
set_truth_lens_model() (h0rton.h0_inference.h0_posterior.H0Posterior method)
(h0rton.h0_inference.H0Posterior method)
setUp() (test_gaussian_bnn_posterior.TestGaussianBNNPosterior method)
(test_h0_posterior.TestH0Posterior method)
(test_h0_utils.TestH0Utils method)
setUpClass() (h0rton.tests.test_trainval_data.test_data_utils.TestDataUtils class method)
(h0rton.tests.test_trainval_data.test_xy_data.TestXYData class method)
(test_bayesian_resnet.TestBayesianResNet class method)
(test_gaussian_bnn_posterior.TestGaussianBNNPosterior class method)
(test_gaussian_nll.TestGaussianNLL class method)
(test_gaussian_nll_cpu.TestGaussianNLLCPU class method)
(test_gaussian_nll_native.TestGaussianNLLNative class method)
(test_h0_posterior.TestH0Posterior class method)
(test_h0_utils.TestH0Utils class method)
(test_mcmc_utils.TestMCMCUtils class method)
(test_plotting_utils.TestPlottingUtils class method)
(test_test_val_cfg.TestTestConfig class method)
(test_train_val_config.TestTrainValConfig class method)
sigmoid() (in module h0rton.h0_inference.gaussian_bnn_posterior_cpu)
slice() (h0rton.losses.BaseGaussianNLL method)
(h0rton.losses.BaseGaussianNLLCPU method)
(h0rton.losses.BaseGaussianNLLNative method)
(h0rton.losses.DiagonalGaussianNLL method)
(h0rton.losses.DiagonalGaussianNLLCPU method)
(h0rton.losses.DiagonalGaussianNLLNative method)
(h0rton.losses.DoubleGaussianNLL method)
(h0rton.losses.DoubleGaussianNLLCPU method)
(h0rton.losses.DoubleGaussianNLLNative method)
(h0rton.losses.DoubleLowRankGaussianNLL method)
(h0rton.losses.DoubleLowRankGaussianNLLNative method)
(h0rton.losses.FullRankGaussianNLL method)
(h0rton.losses.FullRankGaussianNLLCPU method)
(h0rton.losses.FullRankGaussianNLLNative method)
(h0rton.losses.LowRankGaussianNLL method)
(h0rton.losses.LowRankGaussianNLLNative method)
(h0rton.losses.gaussian_nll.BaseGaussianNLL method)
(h0rton.losses.gaussian_nll.DiagonalGaussianNLL method)
(h0rton.losses.gaussian_nll.DoubleGaussianNLL method)
(h0rton.losses.gaussian_nll.DoubleLowRankGaussianNLL method)
(h0rton.losses.gaussian_nll.FullRankGaussianNLL method)
(h0rton.losses.gaussian_nll.LowRankGaussianNLL method)
(h0rton.losses.gaussian_nll_cpu.BaseGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_cpu.DiagonalGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_cpu.DoubleGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_cpu.FullRankGaussianNLLCPU method)
(h0rton.losses.gaussian_nll_native.BaseGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.DiagonalGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.DoubleGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.DoubleLowRankGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.FullRankGaussianNLLNative method)
(h0rton.losses.gaussian_nll_native.LowRankGaussianNLLNative method)
split_component_param() (in module h0rton.h0_inference.mcmc_utils)
sub_name (in module corner_metric_simplified)
subnames (in module corner_metric_simplified)
summarize_mcmc() (in module h0rton.summarize)
summarize_simple_mc_default() (in module h0rton.summarize)
T
TDLMCData (class in h0rton.trainval_data)
(class in h0rton.trainval_data.tdlmc_data)
tearDownClass() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData class method)
test_activation_maps_layer3() (test_bayesian_resnet.TestBayesianResNet method)
test_activation_maps_layer4() (test_bayesian_resnet.TestBayesianResNet method)
test_asinh() (h0rton.tests.test_trainval_data.test_data_utils.TestDataUtils method)
test_bayesian_resnet (module)
test_bayesian_resnet_dropout() (test_bayesian_resnet.TestBayesianResNet method)
test_bayesian_resnet_forward() (test_bayesian_resnet.TestBayesianResNet method)
test_chuck_images() (test_h0_posterior.TestH0Posterior method)
test_combine_lenses() (test_h0_utils.TestH0Utils method)
test_CosmoConverter() (test_h0_utils.TestH0Utils method)
test_diagonal_gaussian_bnn_posterior() (test_gaussian_bnn_posterior.TestGaussianBNNPosterior method)
test_diagonal_gaussian_bnn_posterior_cpu() (test_gaussian_bnn_posterior_cpu.TestGaussianBNNPosteriorCPU method)
test_diagonal_gaussian_nll() (test_gaussian_nll.TestGaussianNLL method)
test_diagonal_gaussian_nll_cpu() (test_gaussian_nll_cpu.TestGaussianNLLCPU method)
test_diagonal_gaussian_nll_native() (test_gaussian_nll_native.TestGaussianNLLNative method)
test_dict_to_array() (test_mcmc_utils.TestMCMCUtils method)
test_double_gaussian_bnn_posterior() (test_gaussian_bnn_posterior.TestGaussianBNNPosterior method)
test_double_gaussian_bnn_posterior_cpu() (test_gaussian_bnn_posterior_cpu.TestGaussianBNNPosteriorCPU method)
test_double_gaussian_nll() (test_gaussian_nll.TestGaussianNLL method)
test_double_gaussian_nll_cpu() (test_gaussian_nll_cpu.TestGaussianNLLCPU method)
test_double_gaussian_nll_native() (test_gaussian_nll_native.TestGaussianNLLNative method)
test_double_low_rank_gaussian_bnn_posterior() (test_gaussian_bnn_posterior.TestGaussianBNNPosterior method)
test_double_low_rank_gaussian_bnn_posterior_cpu() (test_gaussian_bnn_posterior_cpu.TestGaussianBNNPosteriorCPU method)
test_double_low_rank_gaussian_nll() (test_gaussian_nll.TestGaussianNLL method)
test_double_low_rank_gaussian_nll_native() (test_gaussian_nll_native.TestGaussianNLLNative method)
test_full_rank_gaussian_bnn_posterior() (test_gaussian_bnn_posterior.TestGaussianBNNPosterior method)
test_full_rank_gaussian_bnn_posterior_cpu() (test_gaussian_bnn_posterior_cpu.TestGaussianBNNPosteriorCPU method)
test_full_rank_gaussian_nll() (test_gaussian_nll.TestGaussianNLL method)
test_full_rank_gaussian_nll_cpu() (test_gaussian_nll_cpu.TestGaussianNLLCPU method)
test_full_rank_gaussian_nll_native() (test_gaussian_nll_native.TestGaussianNLLNative method)
test_gaussian_bnn_posterior (module)
test_gaussian_bnn_posterior_cpu (module)
test_gaussian_ll_pdf() (test_h0_utils.TestH0Utils method)
test_gaussian_nll (module)
test_gaussian_nll_cpu (module)
test_gaussian_nll_native (module)
test_get_idx_for_params() (test_mcmc_utils.TestMCMCUtils method)
test_get_lens_kwargs() (test_mcmc_utils.TestMCMCUtils method)
test_get_light_kwargs() (test_mcmc_utils.TestMCMCUtils method)
test_get_lognormal_stats() (test_h0_utils.TestH0Utils method)
test_get_lognormal_stats_naive() (test_h0_utils.TestH0Utils method)
test_get_normal_stats() (test_h0_utils.TestH0Utils method)
test_get_normal_stats_naive() (test_h0_utils.TestH0Utils method)
test_get_ps_kwargs() (test_mcmc_utils.TestMCMCUtils method)
test_get_ps_kwargs_src_plane() (test_mcmc_utils.TestMCMCUtils method)
test_get_special_kwargs() (test_mcmc_utils.TestMCMCUtils method)
test_h0_posterior (module)
test_h0_utils (module)
test_H0Posterior_format_lens_model() (test_h0_posterior.TestH0Posterior method)
test_H0Posterior_H0_recovery_true_kappa_time_delays_lens_model() (test_h0_posterior.TestH0Posterior method)
test_H0Posterior_H0_sampling() (test_h0_posterior.TestH0Posterior method)
test_H0Posterior_original_kappa_sampling() (test_h0_posterior.TestH0Posterior method)
test_H0Posterior_transformed_kappa_sampling() (test_h0_posterior.TestH0Posterior method)
test_HybridBNNPenalty() (test_mcmc_utils.TestMCMCUtils method)
test_low_rank_gaussian_bnn_posterior() (test_gaussian_bnn_posterior.TestGaussianBNNPosterior method)
test_low_rank_gaussian_bnn_posterior_cpu() (test_gaussian_bnn_posterior_cpu.TestGaussianBNNPosteriorCPU method)
test_low_rank_gaussian_nll() (test_gaussian_nll.TestGaussianNLL method)
test_low_rank_gaussian_nll_native() (test_gaussian_nll_native.TestGaussianNLLNative method)
test_mcmc_utils (module)
test_plotting_utils (module)
test_plus_1_log() (h0rton.tests.test_trainval_data.test_data_utils.TestDataUtils method)
test_postprocess_mcmc_chain() (test_mcmc_utils.TestMCMCUtils method)
test_pred_to_natural_gaussian() (test_h0_utils.TestH0Utils method)
test_remove_outliers_from_lognormal() (test_h0_utils.TestH0Utils method)
test_remove_parameters_from_pred() (test_mcmc_utils.TestMCMCUtils method)
test_reorder_to_param_class() (test_mcmc_utils.TestMCMCUtils method)
test_reorder_to_tdlmc() (test_h0_utils.TestH0Utils method)
test_rescale_01() (h0rton.tests.test_trainval_data.test_data_utils.TestDataUtils method)
test_split_component_param() (test_mcmc_utils.TestMCMCUtils method)
test_tensor_type() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_test_config_constructor() (test_test_val_cfg.TestTestConfig method)
test_test_val_cfg (module)
test_train_val_absent() (test_train_val_config.TestTrainValConfig method)
test_train_val_config (module)
test_train_val_config_constructor() (test_train_val_config.TestTrainValConfig method)
test_train_vs_val() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_whiten_pixels() (h0rton.tests.test_trainval_data.test_data_utils.TestDataUtils method)
test_whiten_Y_cols() (h0rton.tests.test_trainval_data.test_data_utils.TestDataUtils method)
test_X_exposure_time_factor() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_X_identity() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_X_transformation_log() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_X_transformation_log_rescale() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_X_transformation_rescale() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
test_Y_transformation_() (h0rton.tests.test_trainval_data.test_xy_data.TestXYData method)
TestBayesianResNet (class in test_bayesian_resnet)
TestConfig (class in h0rton.configs)
(class in h0rton.configs.test_config)
TestDataUtils (class in h0rton.tests.test_trainval_data.test_data_utils)
TestGaussianBNNPosterior (class in test_gaussian_bnn_posterior)
TestGaussianBNNPosteriorCPU (class in test_gaussian_bnn_posterior_cpu)
TestGaussianNLL (class in test_gaussian_nll)
TestGaussianNLLCPU (class in test_gaussian_nll_cpu)
TestGaussianNLLNative (class in test_gaussian_nll_native)
TestH0Posterior (class in test_h0_posterior)
TestH0Utils (class in test_h0_utils)
TestMCMCUtils (class in test_mcmc_utils)
TestPlottingUtils (class in test_plotting_utils)
TestTestConfig (class in test_test_val_cfg)
TestTrainValConfig (class in test_train_val_config)
TestXYData (class in h0rton.tests.test_trainval_data.test_xy_data)
TrainValConfig (class in h0rton.configs)
(class in h0rton.configs.train_val_config)
transform_back_cov_mat() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
transform_back_logvar() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
transform_back_mu() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
U
unwhiten_back() (h0rton.h0_inference.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior.BaseGaussianBNNPosterior method)
(h0rton.h0_inference.gaussian_bnn_posterior_cpu.BaseGaussianBNNPosteriorCPU method)
V
validate_user_definition() (h0rton.configs.train_val_config.TrainValConfig method)
(h0rton.configs.TrainValConfig method)
version_dir (in module corner_metric_simplified)
W
whiten_pixels() (in module h0rton.trainval_data.data_utils)
whiten_Y_cols() (in module h0rton.trainval_data.data_utils)
X
XYData (class in h0rton.trainval_data)
(class in h0rton.trainval_data.xy_data)
Y
y_j (in module corner_metric_simplified)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds
Free document hosting provided by
Read the Docs
.