Index _ | A | B | C | D | E | F | G | I | K | L | M | N | O | P | Q | R | S | T | U | V | Z _ __init__() (lightning_uq_box.datamodules.ToyHeteroscedasticDatamodule method) (lightning_uq_box.datamodules.ToyImageRegressionDatamodule method) (lightning_uq_box.models.MLP method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_BatchedRegression method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation method) (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.cards.CARDClassification method) (lightning_uq_box.uq_methods.cards.NoiseScheduler method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemble method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleClassification method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemblePxRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleSegmentation method) (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLRegression method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelRegression method) (lightning_uq_box.uq_methods.deterministic_uncertainty_estimation.DUEClassification method) (lightning_uq_box.uq_methods.deterministic_uncertainty_estimation.DUERegression method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTAClassification method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTARegression method) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase method) (lightning_uq_box.uq_methods.laplace_model.LaplaceClassification method) (lightning_uq_box.uq_methods.laplace_model.LaplaceRegression method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutBase method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVEBase method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVERegression method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegressionBase method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.sgld.SGLD method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.sngp.SNGPClassification method) (lightning_uq_box.uq_methods.swag.SWAGBase method) (lightning_uq_box.uq_methods.swag.SWAGClassification method) (lightning_uq_box.uq_methods.swag.SWAGPxRegression method) (lightning_uq_box.uq_methods.swag.SWAGRegression method) (lightning_uq_box.uq_methods.swag.SWAGSegmentation method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) (lightning_uq_box.uq_methods.zigzag.ZigZagBase method) (lightning_uq_box.uq_methods.zigzag.ZigZagClassification method) A activate_dropout() (lightning_uq_box.uq_methods.mc_dropout.MCDropoutBase method) adapt_output_for_metrics() (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification method) (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelRegression method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVERegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegression method) (lightning_uq_box.uq_methods.swag.SWAGClassification method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) adjust_model_logits() (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) B BNN_LV_VI_Base (class in lightning_uq_box.uq_methods.bnn_lv_vi) BNN_LV_VI_Batched_Base (class in lightning_uq_box.uq_methods.bnn_lv_vi) BNN_LV_VI_Batched_Regression (class in lightning_uq_box.uq_methods.bnn_lv_vi) BNN_LV_VI_Regression (class in lightning_uq_box.uq_methods.bnn_lv_vi) BNN_VI_Base (class in lightning_uq_box.uq_methods.bnn_vi) BNN_VI_BatchedRegression (class in lightning_uq_box.uq_methods.bnn_vi) BNN_VI_ELBO_Base (class in lightning_uq_box.uq_methods.bnn_vi_elbo) BNN_VI_ELBO_Classification (class in lightning_uq_box.uq_methods.bnn_vi_elbo) BNN_VI_ELBO_Regression (class in lightning_uq_box.uq_methods.bnn_vi_elbo) BNN_VI_ELBO_Segmentation (class in lightning_uq_box.uq_methods.bnn_vi_elbo) BNN_VI_Regression (class in lightning_uq_box.uq_methods.bnn_vi) build_model() (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) C calib_dataloader() (lightning_uq_box.datamodules.ToyHeteroscedasticDatamodule method) (lightning_uq_box.datamodules.ToyImageRegressionDatamodule method) CARDBase (class in lightning_uq_box.uq_methods.cards) CARDClassification (class in lightning_uq_box.uq_methods.cards) CARDRegression (class in lightning_uq_box.uq_methods.cards) check_input_layer() (lightning_uq_box.uq_methods.zigzag.ZigZagBase method) compute_aleatoric_uct() (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) compute_elbo_loss() (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) compute_energy_loss() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_BatchedRegression method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Regression method) compute_epistemic_uct() (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) compute_kl_divergence() (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) compute_loss() (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) compute_metrics() (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) compute_predictive_uncertainty() (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTAClassification method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTARegression method) compute_task_loss() (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Regression method) configure_optimizers() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.swag.SWAGBase method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) ConformalQR (class in lightning_uq_box.uq_methods.conformal_qr) constant_schedule() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) convert_to_masked_ensemble() (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) cosine_anneal_schedule() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) cosine_schedule() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) D DeepEnsemble (class in lightning_uq_box.uq_methods.deep_ensemble) DeepEnsembleClassification (class in lightning_uq_box.uq_methods.deep_ensemble) DeepEnsemblePxRegression (class in lightning_uq_box.uq_methods.deep_ensemble) DeepEnsembleRegression (class in lightning_uq_box.uq_methods.deep_ensemble) DeepEnsembleSegmentation (class in lightning_uq_box.uq_methods.deep_ensemble) DensityLayerModelBase (class in lightning_uq_box.uq_methods.density_uncertainty) DensityLayerModelClassification (class in lightning_uq_box.uq_methods.density_uncertainty) DensityLayerModelRegression (class in lightning_uq_box.uq_methods.density_uncertainty) DER (class in lightning_uq_box.uq_methods.deep_evidential_regression) DERPxRegression (class in lightning_uq_box.uq_methods.deep_evidential_regression) diffusion_process() (lightning_uq_box.uq_methods.cards.CARDBase method) DKLBase (class in lightning_uq_box.uq_methods.deep_kernel_learning) DKLClassification (class in lightning_uq_box.uq_methods.deep_kernel_learning) DKLRegression (class in lightning_uq_box.uq_methods.deep_kernel_learning) DUEClassification (class in lightning_uq_box.uq_methods.deterministic_uncertainty_estimation) DUERegression (class in lightning_uq_box.uq_methods.deterministic_uncertainty_estimation) E exclude_from_wt_decay() (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) extract() (lightning_uq_box.uq_methods.cards.CARDBase method) F forward() (lightning_uq_box.models.MLP method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_BatchedRegression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemble method) (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.laplace_model.LaplaceClassification method) (lightning_uq_box.uq_methods.laplace_model.LaplaceRegression method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.zigzag.ZigZagBase method) freeze_layers() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_BatchedRegression method) freeze_model() (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation method) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) G gather_loglikelihood() (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) generate_ensemble_predictions() (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemble method) get_noisy_x_at_t() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) get_rcps_losses_from_outputs() (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) gt_dataloader() (lightning_uq_box.datamodules.ToyHeteroscedasticDatamodule method) I Img2ImgConformal (class in lightning_uq_box.uq_methods.img2img_conformal) K kernel_choices (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase attribute) kl_divergence() (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) L LaplaceBase (class in lightning_uq_box.uq_methods.laplace_model) LaplaceClassification (class in lightning_uq_box.uq_methods.laplace_model) LaplaceRegression (class in lightning_uq_box.uq_methods.laplace_model) lightning_uq_box module lightning_uq_box.datamodules module lightning_uq_box.models module linear_schedule() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) lv_intro_options (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base attribute) M MasksemblesBase (class in lightning_uq_box.uq_methods.masked_ensemble) MasksemblesClassification (class in lightning_uq_box.uq_methods.masked_ensemble) MasksemblesRegression (class in lightning_uq_box.uq_methods.masked_ensemble) MCDropoutBase (class in lightning_uq_box.uq_methods.mc_dropout) MCDropoutClassification (class in lightning_uq_box.uq_methods.mc_dropout) MCDropoutPxRegression (class in lightning_uq_box.uq_methods.mc_dropout) MCDropoutRegression (class in lightning_uq_box.uq_methods.mc_dropout) MCDropoutSegmentation (class in lightning_uq_box.uq_methods.mc_dropout) mean_field_logits() (lightning_uq_box.uq_methods.sngp.SNGPClassification method) metrics_file_name (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal attribute) MLP (class in lightning_uq_box.models) module lightning_uq_box lightning_uq_box.datamodules lightning_uq_box.models MVEBase (class in lightning_uq_box.uq_methods.mean_variance_estimation) MVERegression (class in lightning_uq_box.uq_methods.mean_variance_estimation) N nll_loss (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Regression attribute) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Regression attribute) NoiseScheduler (class in lightning_uq_box.uq_methods.cards) num_input_features (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase property) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase property) num_outputs (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase property) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase property) O on_after_backward() (lightning_uq_box.uq_methods.cards.CARDBase method) on_fit_start() (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) on_test_batch_end() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Regression method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Regression method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation method) (lightning_uq_box.uq_methods.cards.CARDClassification method) (lightning_uq_box.uq_methods.cards.CARDRegression method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleClassification method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemblePxRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleSegmentation method) (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLRegression method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelRegression method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTAClassification method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTARegression method) (lightning_uq_box.uq_methods.laplace_model.LaplaceClassification method) (lightning_uq_box.uq_methods.laplace_model.LaplaceRegression method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVERegression method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegression method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.sngp.SNGPClassification method) (lightning_uq_box.uq_methods.sngp.SNGPRegression method) (lightning_uq_box.uq_methods.swag.SWAGClassification method) (lightning_uq_box.uq_methods.swag.SWAGPxRegression method) (lightning_uq_box.uq_methods.swag.SWAGRegression method) (lightning_uq_box.uq_methods.swag.SWAGSegmentation method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) (lightning_uq_box.uq_methods.zigzag.ZigZagClassification method) (lightning_uq_box.uq_methods.zigzag.ZigZagRegression method) on_test_end() (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) on_test_epoch_end() (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLRegression method) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) on_test_start() (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemblePxRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleSegmentation method) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase method) (lightning_uq_box.uq_methods.laplace_model.LaplaceRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.swag.SWAGPxRegression method) (lightning_uq_box.uq_methods.swag.SWAGSegmentation method) on_train_end() (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.swag.SWAGBase method) on_train_epoch_end() (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.swag.SWAGBase method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) on_train_epoch_start() (lightning_uq_box.uq_methods.sngp.SNGPBase method) on_validation_epoch_end() (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.swag.SWAGBase method) on_validation_start() (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) P p_sample() (lightning_uq_box.uq_methods.cards.CARDBase method) p_sample_loop() (lightning_uq_box.uq_methods.cards.CARDBase method) p_sample_t_1to0() (lightning_uq_box.uq_methods.cards.CARDBase method) pred_dir_name (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation attribute) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemblePxRegression attribute) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleSegmentation attribute) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression attribute) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal attribute) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression attribute) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation attribute) (lightning_uq_box.uq_methods.prob_unet.ProbUNet attribute) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression attribute) (lightning_uq_box.uq_methods.swag.SWAGPxRegression attribute) (lightning_uq_box.uq_methods.swag.SWAGSegmentation attribute) pred_file_name (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Regression attribute) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Regression attribute) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base attribute) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Regression attribute) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification attribute) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Regression attribute) (lightning_uq_box.uq_methods.cards.CARDBase attribute) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR attribute) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleClassification attribute) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleRegression attribute) (lightning_uq_box.uq_methods.deep_evidential_regression.DER attribute) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase attribute) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase attribute) (lightning_uq_box.uq_methods.inference_time_augmentation.TTAClassification attribute) (lightning_uq_box.uq_methods.inference_time_augmentation.TTARegression attribute) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase attribute) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification attribute) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesRegression attribute) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification attribute) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression attribute) (lightning_uq_box.uq_methods.mean_variance_estimation.MVERegression attribute) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegression attribute) (lightning_uq_box.uq_methods.raps.RAPS attribute) (lightning_uq_box.uq_methods.sngp.SNGPBase attribute) (lightning_uq_box.uq_methods.swag.SWAGClassification attribute) (lightning_uq_box.uq_methods.swag.SWAGRegression attribute) (lightning_uq_box.uq_methods.temp_scaling.TempScaling attribute) (lightning_uq_box.uq_methods.zigzag.ZigZagClassification attribute) (lightning_uq_box.uq_methods.zigzag.ZigZagRegression attribute) predict_step() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_BatchedRegression method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation method) (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.cards.CARDClassification method) (lightning_uq_box.uq_methods.cards.CARDRegression method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleClassification method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleSegmentation method) (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLRegression method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelRegression method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) (lightning_uq_box.uq_methods.laplace_model.LaplaceClassification method) (lightning_uq_box.uq_methods.laplace_model.LaplaceRegression method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVERegression method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegression method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.sngp.SNGPClassification method) (lightning_uq_box.uq_methods.swag.SWAGClassification method) (lightning_uq_box.uq_methods.swag.SWAGRegression method) (lightning_uq_box.uq_methods.swag.SWAGSegmentation method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) (lightning_uq_box.uq_methods.zigzag.ZigZagClassification method) (lightning_uq_box.uq_methods.zigzag.ZigZagRegression method) ProbUNet (class in lightning_uq_box.uq_methods.prob_unet) Q q_sample() (lightning_uq_box.uq_methods.cards.CARDBase method) quadratic_schedule() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) QuantilePxRegression (class in lightning_uq_box.uq_methods.quantile_regression) QuantileRegression (class in lightning_uq_box.uq_methods.quantile_regression) QuantileRegressionBase (class in lightning_uq_box.uq_methods.quantile_regression) R RAPS (class in lightning_uq_box.uq_methods.raps) rcps_loss_fn() (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) recompute_covariance_matrix() (lightning_uq_box.uq_methods.sngp.SNGPBase method) reconstruct() (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) reset_precision_matrix() (lightning_uq_box.uq_methods.sngp.SNGPBase method) S sample() (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) sample_latent_variable_prior() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Base method) sample_predictions() (lightning_uq_box.uq_methods.swag.SWAGBase method) sample_state() (lightning_uq_box.uq_methods.swag.SWAGBase method) setup_task() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Batched_Regression method) (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Regression method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Regression method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Segmentation method) (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.cards.CARDClassification method) (lightning_uq_box.uq_methods.cards.CARDRegression method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemble method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleClassification method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemblePxRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleRegression method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleSegmentation method) (lightning_uq_box.uq_methods.deep_evidential_regression.DER method) (lightning_uq_box.uq_methods.deep_evidential_regression.DERPxRegression method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLRegression method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelRegression method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTAClassification method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTARegression method) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase method) (lightning_uq_box.uq_methods.laplace_model.LaplaceClassification method) (lightning_uq_box.uq_methods.laplace_model.LaplaceRegression method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutBase method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutPxRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutSegmentation method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVEBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegressionBase method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.sngp.SNGPClassification method) (lightning_uq_box.uq_methods.sngp.SNGPRegression method) (lightning_uq_box.uq_methods.swag.SWAGBase method) (lightning_uq_box.uq_methods.swag.SWAGClassification method) (lightning_uq_box.uq_methods.swag.SWAGPxRegression method) (lightning_uq_box.uq_methods.swag.SWAGRegression method) (lightning_uq_box.uq_methods.swag.SWAGSegmentation method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.zigzag.ZigZagClassification method) (lightning_uq_box.uq_methods.zigzag.ZigZagRegression method) SGLD (class in lightning_uq_box.uq_methods.sgld) sigmoid_schedule() (lightning_uq_box.uq_methods.cards.NoiseScheduler method) SNGPBase (class in lightning_uq_box.uq_methods.sngp) SNGPClassification (class in lightning_uq_box.uq_methods.sngp) SNGPRegression (class in lightning_uq_box.uq_methods.sngp) step() (lightning_uq_box.uq_methods.sgld.SGLD method) SWAGBase (class in lightning_uq_box.uq_methods.swag) SWAGClassification (class in lightning_uq_box.uq_methods.swag) SWAGPxRegression (class in lightning_uq_box.uq_methods.swag) SWAGRegression (class in lightning_uq_box.uq_methods.swag) SWAGSegmentation (class in lightning_uq_box.uq_methods.swag) T TempScaling (class in lightning_uq_box.uq_methods.temp_scaling) test_dataloader() (lightning_uq_box.datamodules.ToyHeteroscedasticDatamodule method) (lightning_uq_box.datamodules.ToyImageRegressionDatamodule method) test_step() (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.conformal_qr.ConformalQR method) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsemble method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) (lightning_uq_box.uq_methods.img2img_conformal.Img2ImgConformal method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) (lightning_uq_box.uq_methods.laplace_model.LaplaceBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.quantile_regression.QuantilePxRegression method) (lightning_uq_box.uq_methods.quantile_regression.QuantileRegression method) (lightning_uq_box.uq_methods.raps.RAPS method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.temp_scaling.TempScaling method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) (lightning_uq_box.uq_methods.zigzag.ZigZagBase method) ToyHeteroscedasticDatamodule (class in lightning_uq_box.datamodules) ToyImageRegressionDatamodule (class in lightning_uq_box.datamodules) train_dataloader() (lightning_uq_box.datamodules.ToyHeteroscedasticDatamodule method) (lightning_uq_box.datamodules.ToyImageRegressionDatamodule method) training_step() (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutBase method) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutRegression method) (lightning_uq_box.uq_methods.mean_variance_estimation.MVEBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.swag.SWAGBase method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) (lightning_uq_box.uq_methods.zigzag.ZigZagBase method) TTABase (class in lightning_uq_box.uq_methods.inference_time_augmentation) TTAClassification (class in lightning_uq_box.uq_methods.inference_time_augmentation) TTARegression (class in lightning_uq_box.uq_methods.inference_time_augmentation) U unfreeze_layers() (lightning_uq_box.uq_methods.bnn_lv_vi.BNN_LV_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) update_uncertainty_buffers() (lightning_uq_box.uq_methods.swag.SWAGBase method) V val_dataloader() (lightning_uq_box.datamodules.ToyHeteroscedasticDatamodule method) (lightning_uq_box.datamodules.ToyImageRegressionDatamodule method) valid_lambda_criterion (lightning_uq_box.uq_methods.raps.RAPS attribute) valid_layer_types (lightning_uq_box.uq_methods.vbll.VBLLClassification attribute) valid_merge_strategies (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase attribute) valid_schedules (lightning_uq_box.uq_methods.cards.NoiseScheduler attribute) valid_tasks (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Classification attribute) (lightning_uq_box.uq_methods.cards.CARDClassification attribute) (lightning_uq_box.uq_methods.deep_ensemble.DeepEnsembleClassification attribute) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification attribute) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelClassification attribute) (lightning_uq_box.uq_methods.inference_time_augmentation.TTAClassification attribute) (lightning_uq_box.uq_methods.laplace_model.LaplaceClassification attribute) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesClassification attribute) (lightning_uq_box.uq_methods.mc_dropout.MCDropoutClassification attribute) (lightning_uq_box.uq_methods.prob_unet.ProbUNet attribute) (lightning_uq_box.uq_methods.raps.RAPS attribute) (lightning_uq_box.uq_methods.sngp.SNGPClassification attribute) (lightning_uq_box.uq_methods.swag.SWAGClassification attribute) (lightning_uq_box.uq_methods.temp_scaling.TempScaling attribute) (lightning_uq_box.uq_methods.zigzag.ZigZagClassification attribute) validation_step() (lightning_uq_box.uq_methods.bnn_vi.BNN_VI_Base method) (lightning_uq_box.uq_methods.bnn_vi_elbo.BNN_VI_ELBO_Base method) (lightning_uq_box.uq_methods.cards.CARDBase method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLBase method) (lightning_uq_box.uq_methods.deep_kernel_learning.DKLClassification method) (lightning_uq_box.uq_methods.density_uncertainty.DensityLayerModelBase method) (lightning_uq_box.uq_methods.inference_time_augmentation.TTABase method) (lightning_uq_box.uq_methods.masked_ensemble.MasksemblesBase method) (lightning_uq_box.uq_methods.prob_unet.ProbUNet method) (lightning_uq_box.uq_methods.sngp.SNGPBase method) (lightning_uq_box.uq_methods.swag.SWAGBase method) (lightning_uq_box.uq_methods.vbll.VBLLClassification method) (lightning_uq_box.uq_methods.vbll.VBLLRegression method) (lightning_uq_box.uq_methods.zigzag.ZigZagBase method) VBLLClassification (class in lightning_uq_box.uq_methods.vbll) VBLLRegression (class in lightning_uq_box.uq_methods.vbll) Z ZigZagBase (class in lightning_uq_box.uq_methods.zigzag) ZigZagClassification (class in lightning_uq_box.uq_methods.zigzag) ZigZagRegression (class in lightning_uq_box.uq_methods.zigzag)
