Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W A accuracy() (in module fortuna.metric.classification) activation (fortuna.model.resnet.ResNet attribute) (fortuna.model.wideresnet.WideResNet attribute) activations (fortuna.model.mlp.MLP attribute) AdaptiveConformalClassifier (class in fortuna.conformal.classification.adaptive_conformal_classifier) AdaptiveConformalRegressor (class in fortuna.conformal.regression.adaptive_conformal_regressor) AdaptivePredictionConformalClassifier (class in fortuna.conformal.classification.adaptive_prediction) ADVIPosterior (class in fortuna.prob_model.posterior.normalizing_flow.advi.advi_posterior) ADVIPosteriorApproximator (class in fortuna.prob_model.posterior.normalizing_flow.advi.advi_approximator) ADVIState (class in fortuna.prob_model.posterior.normalizing_flow.advi.advi_state) aleatoric_entropy() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) aleatoric_variance() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) apply() (fortuna.model.model_manager.base.ModelManager method) (fortuna.model.model_manager.classification.ClassificationModelManager method) (fortuna.model.model_manager.classification.SNGPClassificationModelManager method) (fortuna.model.model_manager.regression.RegressionModelManager method) (fortuna.model_editor.base.ModelEditor method) apply_patches() (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) Array (in module fortuna.typing) B Batch (in module fortuna.typing) BatchMVPConformalRegressor (class in fortuna.conformal.regression.batch_mvp) bias_init (fortuna.model.utils.random_features.RandomFourierFeatures attribute) bind() (fortuna.model_editor.base.ModelEditor method) block_cls (fortuna.model.resnet.ResNet attribute) brier_score() (in module fortuna.metric.classification) bucket_types (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor property) C CalibCheckpointer (class in fortuna.prob_model.calib_config.checkpointer) CalibConfig (class in fortuna.prob_model.calib_config.base) CalibMonitor (class in fortuna.prob_model.calib_config.monitor) CalibMutable (in module fortuna.typing) CalibOptimizer (class in fortuna.prob_model.calib_config.optimizer) CalibParams (in module fortuna.typing) CalibProcessor (class in fortuna.prob_model.calib_config.processor) calibrate() (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) (fortuna.output_calib_model.classification.OutputCalibClassifier method) (fortuna.output_calib_model.regression.OutputCalibRegressor method) (fortuna.prob_model.classification.ProbClassifier method) (fortuna.prob_model.regression.ProbRegressor method) calibration_error() (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) Callback (class in fortuna.prob_model.fit_config.callback) (class in fortuna.training.callback) Checkpointer (class in fortuna.output_calib_model.config.checkpointer) chop() (fortuna.data.loader.DataLoader method) (fortuna.data.loader.InputsLoader method) (fortuna.data.loader.TargetsLoader method) ClassificationLikelihood (class in fortuna.likelihood.classification) ClassificationMaskedProbOutputLayer (class in fortuna.prob_output_layer.classification) ClassificationModelManager (class in fortuna.model.model_manager.classification) ClassificationPredictive (class in fortuna.output_calib_model.predictive.classification) (class in fortuna.prob_model.predictive.classification) ClassificationProbOutputLayer (class in fortuna.prob_output_layer.classification) ClassificationTemperatureScaler (class in fortuna.output_calibrator.classification) clone() (fortuna.model_editor.base.ModelEditor method) Config (class in fortuna.output_calib_model.config.base) conformal_interval() (fortuna.conformal.regression.cvplus.CVPlusConformalRegressor method) (fortuna.conformal.regression.enbpi.EnbPI method) (fortuna.conformal.regression.jackknife_minmax.JackknifeMinmaxConformalRegressor method) (fortuna.conformal.regression.jackknifeplus.JackknifePlusConformalRegressor method) (fortuna.conformal.regression.onedim_uncertainty.OneDimensionalUncertaintyConformalRegressor method) (fortuna.conformal.regression.quantile.QuantileConformalRegressor method) conformal_interval_from_residuals() (fortuna.conformal.regression.enbpi.EnbPI method) conformal_set() (fortuna.conformal.classification.adaptive_prediction.AdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.CVPlusAdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.CVPlusSimplePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.SimplePredictionConformalClassifier method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) constant_schedule() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule) constant_schedule_with_cosine_burnin() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule) ConstantModel (class in fortuna.model.constant) conv (fortuna.model.resnet.ResNet attribute) (fortuna.model.wideresnet.WideResNet attribute) convert_from_map_state() (fortuna.prob_model.posterior.laplace.laplace_state.LaplaceState class method) (fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_state.CyclicalSGLDState class method) (fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_state.SGHMCState class method) (fortuna.prob_model.posterior.swag.swag_state.SWAGState class method) cosine_schedule() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule) cov (fortuna.ood_detection.ddu.DeepDeterministicUncertaintyOODClassifier property) (fortuna.ood_detection.mahalanobis.MalahanobisOODClassifier property) covariance_kwargs (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) credible_interval() (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) credible_set() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) CVPlusAdaptivePredictionConformalClassifier (class in fortuna.conformal.classification.adaptive_prediction) CVPlusConformalRegressor (class in fortuna.conformal.regression.cvplus) CVPlusSimplePredictionConformalClassifier (class in fortuna.conformal.classification.simple_prediction) cyclical_cosine_schedule_with_const_burnin() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule) CyclicalSGLDPosterior (class in fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_posterior) CyclicalSGLDPosteriorApproximator (class in fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_approximator) CyclicalSGLDState (class in fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_state) D DataLoader (class in fortuna.data.loader) DeepDeterministicUncertaintyOODClassifier (class in fortuna.ood_detection.ddu) DeepEnsemblePosterior (class in fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_posterior) DeepEnsemblePosteriorApproximator (class in fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_approximator) dense (fortuna.model.mlp.MLP attribute) depth (fortuna.model.wideresnet.WideResNet attribute) dev (fortuna.prob_model.posterior.swag.swag_state.SWAGState attribute) DiagonalGaussianPrior (class in fortuna.prob_model.prior.gaussian) dropout (fortuna.model.mlp.MLP attribute) (fortuna.model.wideresnet.WideResNet attribute) dropout_rate (fortuna.model.mlp.MLP attribute) (fortuna.model.wideresnet.WideResNet attribute) dtype (fortuna.model.lenet.LeNet5 attribute) (fortuna.model.resnet.ResNet attribute) (fortuna.model.utils.random_features.LaplaceRandomFeatureCovariance attribute) (fortuna.model.utils.random_features.RandomFourierFeatures attribute) (fortuna.model.wideresnet.WideResNet attribute) dynamic_scale (fortuna.prob_model.posterior.state.PosteriorState attribute) E ece() (in module fortuna.metric.classification) effective_sample_size() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_diagnostic) EnbPI (class in fortuna.conformal.regression.enbpi) encoded_name (fortuna.prob_model.posterior.laplace.laplace_state.LaplaceState attribute) (fortuna.prob_model.posterior.map.map_state.MAPState attribute) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_state.ADVIState attribute) (fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_state.CyclicalSGLDState attribute) (fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_state.SGHMCState attribute) (fortuna.prob_model.posterior.swag.swag_state.SWAGState attribute) ensemble_log_prob() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) entropy() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) epistemic_entropy() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) epistemic_variance() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) eta (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor property) expected_calibration_error() (in module fortuna.metric.classification) F feature_scale (fortuna.model.utils.random_features.RandomFourierFeatures attribute) features (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) (fortuna.model.utils.random_features.RandomFourierFeatures attribute) fit() (fortuna.ood_detection.ddu.DeepDeterministicUncertaintyOODClassifier method) (fortuna.ood_detection.mahalanobis.MalahanobisOODClassifier method) (fortuna.prob_model.posterior.base.Posterior method) (fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_posterior.DeepEnsemblePosterior method) (fortuna.prob_model.posterior.laplace.laplace_posterior.LaplacePosterior method) (fortuna.prob_model.posterior.map.map_posterior.MAPPosterior method) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_posterior.ADVIPosterior method) (fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_posterior.CyclicalSGLDPosterior method) (fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_posterior.SGHMCPosterior method) (fortuna.prob_model.posterior.sngp.sngp_posterior.SNGPPosterior method) (fortuna.prob_model.posterior.swag.swag_posterior.SWAGPosterior method) FitCheckpointer (class in fortuna.prob_model.fit_config.checkpointer) FitConfig (class in fortuna.prob_model.fit_config.base) FitMonitor (class in fortuna.prob_model.fit_config.monitor) FitOptimizer (class in fortuna.prob_model.fit_config.optimizer) FitProcessor (class in fortuna.prob_model.fit_config.processor) flax.linen.Module module fortuna.conformal.classification.adaptive_conformal_classifier module fortuna.conformal.classification.adaptive_prediction module fortuna.conformal.classification.simple_prediction module fortuna.conformal.regression.adaptive_conformal_regressor module fortuna.conformal.regression.batch_mvp module fortuna.conformal.regression.cvplus module fortuna.conformal.regression.enbpi module fortuna.conformal.regression.jackknife_minmax module fortuna.conformal.regression.jackknifeplus module fortuna.conformal.regression.onedim_uncertainty module fortuna.conformal.regression.quantile module fortuna.likelihood.base module fortuna.likelihood.classification module fortuna.likelihood.regression module fortuna.metric.classification module fortuna.metric.regression module fortuna.model.model_manager.base module fortuna.model.model_manager.regression module fortuna.model_editor.base module fortuna.output_calib_model.base module fortuna.output_calib_model.classification module fortuna.output_calib_model.predictive.classification module fortuna.output_calib_model.predictive.regression module fortuna.output_calib_model.regression module fortuna.output_calibrator.classification module fortuna.output_calibrator.regression module fortuna.prob_model.base module fortuna.prob_model.classification module fortuna.prob_model.joint.base module fortuna.prob_model.posterior.sgmcmc.sgmcmc_diagnostic module fortuna.prob_model.posterior.sgmcmc.sgmcmc_preconditioner module fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule module fortuna.prob_model.predictive.classification module fortuna.prob_model.predictive.regression module fortuna.prob_model.prior.base module fortuna.prob_model.prior.gaussian module fortuna.prob_model.regression module fortuna.prob_output_layer.base module fortuna.prob_output_layer.classification module fortuna.prob_output_layer.regression module from_array_data() (fortuna.data.loader.DataLoader class method) from_array_inputs() (fortuna.data.loader.InputsLoader class method) from_array_targets() (fortuna.data.loader.TargetsLoader class method) from_callable_iterable() (fortuna.data.loader.DataLoader class method) (fortuna.data.loader.InputsLoader class method) (fortuna.data.loader.TargetsLoader class method) from_data_loader() (fortuna.data.loader.InputsLoader class method) (fortuna.data.loader.TargetsLoader class method) from_inputs_loaders() (fortuna.data.loader.DataLoader class method) from_iterable() (fortuna.data.loader.DataLoader class method) (fortuna.data.loader.InputsLoader class method) (fortuna.data.loader.TargetsLoader class method) from_tensorflow_data_loader() (fortuna.data.loader.DataLoader class method) from_torch_data_loader() (fortuna.data.loader.DataLoader class method) G get() (fortuna.utils.random.RandomNumberGenerator method) get_calibrated_outputs() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) get_outputs() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) get_scores() (fortuna.conformal.classification.adaptive_conformal_classifier.AdaptiveConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.AdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.CVPlusAdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.CVPlusSimplePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.SimplePredictionConformalClassifier method) (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) get_variable() (fortuna.model_editor.base.ModelEditor method) grad_accumulated (fortuna.prob_model.posterior.state.PosteriorState attribute) H has_rng() (fortuna.model_editor.base.ModelEditor method) has_variable() (fortuna.model_editor.base.ModelEditor method) hidden_features (fortuna.model.utils.random_features.LaplaceRandomFeatureCovariance attribute) (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) hidden_kwargs (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) HyperparameterModel (class in fortuna.model.hyper) I identity_preconditioner() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_preconditioner) init() (fortuna.model.model_manager.base.ModelManager method) (fortuna.model.model_manager.classification.ClassificationModelManager method) (fortuna.model.model_manager.classification.SNGPClassificationModelManager method) (fortuna.model.model_manager.regression.RegressionModelManager method) (fortuna.model_editor.base.ModelEditor method) (fortuna.output_calib_model.state.OutputCalibState class method) (fortuna.prob_model.joint.base.Joint method) (fortuna.prob_model.posterior.laplace.laplace_state.LaplaceState class method) (fortuna.prob_model.posterior.map.map_state.MAPState class method) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_state.ADVIState class method) (fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_state.CyclicalSGLDState class method) (fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_state.SGHMCState class method) (fortuna.prob_model.posterior.state.PosteriorState class method) (fortuna.prob_model.posterior.swag.swag_state.SWAGState class method) init_from_dict() (fortuna.model.model_manager.state.ModelManagerState class method) (fortuna.output_calib_model.state.OutputCalibState class method) (fortuna.prob_model.posterior.laplace.laplace_state.LaplaceState class method) (fortuna.prob_model.posterior.map.map_state.MAPState class method) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_state.ADVIState class method) (fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_state.CyclicalSGLDState class method) (fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_state.SGHMCState class method) (fortuna.prob_model.posterior.state.PosteriorState class method) (fortuna.prob_model.posterior.swag.swag_state.SWAGState class method) init_from_states() (fortuna.prob_model.joint.state.JointState class method) init_with_output() (fortuna.model_editor.base.ModelEditor method) input_shape (fortuna.data.loader.DataLoader property) (fortuna.data.loader.InputsLoader property) InputsLoader (class in fortuna.data.loader) is_in() (fortuna.conformal.classification.adaptive_conformal_classifier.AdaptiveConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.AdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.CVPlusAdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.CVPlusSimplePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.SimplePredictionConformalClassifier method) (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) (fortuna.conformal.regression.cvplus.CVPlusConformalRegressor method) (fortuna.conformal.regression.enbpi.EnbPI method) (fortuna.conformal.regression.jackknife_minmax.JackknifeMinmaxConformalRegressor method) (fortuna.conformal.regression.jackknifeplus.JackknifePlusConformalRegressor method) (fortuna.conformal.regression.onedim_uncertainty.OneDimensionalUncertaintyConformalRegressor method) (fortuna.conformal.regression.quantile.QuantileConformalRegressor method) is_initializing() (fortuna.model_editor.base.ModelEditor method) is_mutable_collection() (fortuna.model_editor.base.ModelEditor method) IsotropicGaussianPrior (class in fortuna.prob_model.prior.gaussian) iteration (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) J JackknifeMinmaxConformalRegressor (class in fortuna.conformal.regression.jackknife_minmax) JackknifePlusConformalRegressor (class in fortuna.conformal.regression.jackknifeplus) Joint (class in fortuna.prob_model.joint.base) joint (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) JointState (class in fortuna.prob_model.joint.state) K kernel_apply_kwargs (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) kernel_init (fortuna.model.utils.random_features.RandomFourierFeatures attribute) kernel_name (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) kernel_stein_discrepancy_imq() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_diagnostic) L LaplacePosterior (class in fortuna.prob_model.posterior.laplace.laplace_posterior) LaplacePosteriorApproximator (class in fortuna.prob_model.posterior.laplace.laplace_approximator) LaplaceRandomFeatureCovariance (class in fortuna.model.utils.random_features) LaplaceState (class in fortuna.prob_model.posterior.laplace.laplace_state) layer (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) layer_name (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) lazy_init() (fortuna.model_editor.base.ModelEditor method) LeNet5 (class in fortuna.model.lenet) Likelihood (class in fortuna.likelihood.base) likelihood (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) load_state() (fortuna.output_calib_model.base.OutputCalibModel method) (fortuna.output_calib_model.classification.OutputCalibClassifier method) (fortuna.output_calib_model.regression.OutputCalibRegressor method) (fortuna.prob_model.base.ProbModel method) (fortuna.prob_model.classification.ProbClassifier method) (fortuna.prob_model.posterior.base.Posterior method) (fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_posterior.DeepEnsemblePosterior method) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_posterior.ADVIPosterior method) (fortuna.prob_model.regression.ProbRegressor method) log_joint_prob() (fortuna.prob_model.prior.base.Prior method) (fortuna.prob_model.prior.gaussian.DiagonalGaussianPrior method) (fortuna.prob_model.prior.gaussian.IsotropicGaussianPrior method) log_prob() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) M mae() (in module fortuna.metric.regression) make_rng() (fortuna.model_editor.base.ModelEditor method) MalahanobisOODClassifier (class in fortuna.ood_detection.mahalanobis) MAPPosterior (class in fortuna.prob_model.posterior.map.map_posterior) MAPPosteriorApproximator (class in fortuna.prob_model.posterior.map.map_approximator) MAPState (class in fortuna.prob_model.posterior.map.map_state) maximum_calibration_error() (in module fortuna.metric.classification) mce() (in module fortuna.metric.classification) mean (fortuna.ood_detection.ddu.DeepDeterministicUncertaintyOODClassifier property) (fortuna.ood_detection.mahalanobis.MalahanobisOODClassifier property) (fortuna.prob_model.posterior.swag.swag_state.SWAGState attribute) mean() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) mean_absolute_error() (in module fortuna.metric.regression) mean_squared_error() (in module fortuna.metric.regression) MLP (class in fortuna.model.mlp) mode() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) model (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) model_manager (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) ModelEditor (class in fortuna.model_editor.base) ModelManager (class in fortuna.model.model_manager.base) ModelManagerState (class in fortuna.model.model_manager.state) module flax.linen.Module fortuna.conformal.classification.adaptive_conformal_classifier fortuna.conformal.classification.adaptive_prediction fortuna.conformal.classification.simple_prediction fortuna.conformal.regression.adaptive_conformal_regressor fortuna.conformal.regression.batch_mvp fortuna.conformal.regression.cvplus fortuna.conformal.regression.enbpi fortuna.conformal.regression.jackknife_minmax fortuna.conformal.regression.jackknifeplus fortuna.conformal.regression.onedim_uncertainty fortuna.conformal.regression.quantile fortuna.likelihood.base fortuna.likelihood.classification fortuna.likelihood.regression fortuna.metric.classification fortuna.metric.regression fortuna.model.model_manager.base fortuna.model.model_manager.regression fortuna.model_editor.base fortuna.output_calib_model.base fortuna.output_calib_model.classification fortuna.output_calib_model.predictive.classification fortuna.output_calib_model.predictive.regression fortuna.output_calib_model.regression fortuna.output_calibrator.classification fortuna.output_calibrator.regression fortuna.prob_model.base fortuna.prob_model.classification fortuna.prob_model.joint.base fortuna.prob_model.posterior.sgmcmc.sgmcmc_diagnostic fortuna.prob_model.posterior.sgmcmc.sgmcmc_preconditioner fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule fortuna.prob_model.predictive.classification fortuna.prob_model.predictive.regression fortuna.prob_model.prior.base fortuna.prob_model.prior.gaussian fortuna.prob_model.regression fortuna.prob_output_layer.base fortuna.prob_output_layer.classification fortuna.prob_output_layer.regression momentum (fortuna.model.utils.random_features.LaplaceRandomFeatureCovariance attribute) Monitor (class in fortuna.output_calib_model.config.monitor) mse() (in module fortuna.metric.regression) Mutable (in module fortuna.typing) N n_buckets (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor property) name (fortuna.model_editor.base.ModelEditor attribute) (fortuna.output_calibrator.classification.ClassificationTemperatureScaler attribute) (fortuna.output_calibrator.regression.RegressionTemperatureScaler attribute) norm_kwargs (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) norm_multiplier (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) normalize_input (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) num_filters (fortuna.model.resnet.ResNet attribute) num_unique_labels (fortuna.data.loader.DataLoader property) O OneDimensionalUncertaintyConformalRegressor (class in fortuna.conformal.regression.onedim_uncertainty) OptaxOptimizer (in module fortuna.typing) Optimizer (class in fortuna.output_calib_model.config.optimizer) output_calib_manager (fortuna.output_calib_model.classification.OutputCalibClassifier attribute) (fortuna.output_calib_model.regression.OutputCalibRegressor attribute) output_calibrator (fortuna.output_calib_model.classification.OutputCalibClassifier attribute) (fortuna.output_calib_model.regression.OutputCalibRegressor attribute) (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) output_dim (fortuna.model.lenet.LeNet5 attribute) (fortuna.model.mlp.MLP attribute) (fortuna.model.resnet.ResNet attribute) (fortuna.model.wideresnet.WideResNet attribute) output_kwargs (fortuna.model.utils.random_features.RandomFeatureGaussianProcess attribute) OutputCalibClassifier (class in fortuna.output_calib_model.classification) OutputCalibModel (class in fortuna.output_calib_model.base) OutputCalibRegressor (class in fortuna.output_calib_model.regression) OutputCalibState (class in fortuna.output_calib_model.state) P param() (fortuna.model_editor.base.ModelEditor method) Params (in module fortuna.typing) parent (fortuna.model_editor.base.ModelEditor attribute) (fortuna.output_calibrator.classification.ClassificationTemperatureScaler attribute) (fortuna.output_calibrator.regression.RegressionTemperatureScaler attribute) patch_type (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor property) patches (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor property) Path (in module fortuna.typing) perturb() (fortuna.model_editor.base.ModelEditor method) picp() (in module fortuna.metric.regression) pinball_loss() (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) plot_reliability_diagram() (in module fortuna.plot) polynomial_schedule() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_step_schedule) Posterior (class in fortuna.prob_model.posterior.base) posterior (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) posterior_approximator (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) posterior_method_kwargs (fortuna.prob_model.posterior.base.PosteriorApproximator property) (fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_approximator.DeepEnsemblePosteriorApproximator property) (fortuna.prob_model.posterior.laplace.laplace_approximator.LaplacePosteriorApproximator property) (fortuna.prob_model.posterior.map.map_approximator.MAPPosteriorApproximator property) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_approximator.ADVIPosteriorApproximator property) (fortuna.prob_model.posterior.sgmcmc.cyclical_sgld.cyclical_sgld_approximator.CyclicalSGLDPosteriorApproximator property) (fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_approximator.SGHMCPosteriorApproximator property) (fortuna.prob_model.posterior.sngp.sngp_approximator.SNGPPosteriorApproximator property) (fortuna.prob_model.posterior.swag.swag_approximator.SWAGPosteriorApproximator property) PosteriorApproximator (class in fortuna.prob_model.posterior.base) PosteriorState (class in fortuna.prob_model.posterior.state) predict() (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) prediction_interval_coverage_probability() (in module fortuna.metric.regression) predictive (fortuna.output_calib_model.classification.OutputCalibClassifier attribute) (fortuna.output_calib_model.regression.OutputCalibRegressor attribute) (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) Prior (class in fortuna.prob_model.prior.base) prior (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) prior_log_var (fortuna.prob_model.posterior.laplace.laplace_state.LaplaceState attribute) prior_log_variance_tuning() (fortuna.prob_model.posterior.laplace.laplace_posterior.LaplacePosterior method) prob_output_layer (fortuna.output_calib_model.classification.OutputCalibClassifier attribute) (fortuna.output_calib_model.regression.OutputCalibRegressor attribute) (fortuna.prob_model.classification.ProbClassifier attribute) (fortuna.prob_model.regression.ProbRegressor attribute) ProbClassifier (class in fortuna.prob_model.classification) ProbModel (class in fortuna.prob_model.base) ProbOutputLayer (class in fortuna.prob_output_layer.base) ProbRegressor (class in fortuna.prob_model.regression) Processor (class in fortuna.output_calib_model.config.processor) put_variable() (fortuna.model_editor.base.ModelEditor method) Q quantile() (fortuna.conformal.regression.onedim_uncertainty.OneDimensionalUncertaintyConformalRegressor method) (fortuna.conformal.regression.quantile.QuantileConformalRegressor method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) QuantileConformalRegressor (class in fortuna.conformal.regression.quantile) R RandomFeatureGaussianProcess (class in fortuna.model.utils.random_features) RandomFourierFeatures (class in fortuna.model.utils.random_features) RandomNumberGenerator (class in fortuna.utils.random) RegressionLikelihood (class in fortuna.likelihood.regression) RegressionModelManager (class in fortuna.model.model_manager.regression) RegressionPredictive (class in fortuna.output_calib_model.predictive.regression) (class in fortuna.prob_model.predictive.regression) RegressionProbOutputLayer (class in fortuna.prob_output_layer.regression) RegressionTemperatureScaler (class in fortuna.output_calibrator.regression) ResetCovarianceCallback (class in fortuna.prob_model.posterior.sngp.sngp_callback) ResNet (class in fortuna.model.resnet) ResNet101 (in module fortuna.model.resnet) ResNet152 (in module fortuna.model.resnet) ResNet18 (in module fortuna.model.resnet) ResNet200 (in module fortuna.model.resnet) ResNet34 (in module fortuna.model.resnet) ResNet50 (in module fortuna.model.resnet) ridge_penalty (fortuna.model.utils.random_features.LaplaceRandomFeatureCovariance attribute) rmae() (in module fortuna.metric.regression) rmse() (in module fortuna.metric.regression) rmsprop_preconditioner() (in module fortuna.prob_model.posterior.sgmcmc.sgmcmc_preconditioner) rng (fortuna.likelihood.base.Likelihood property) (fortuna.likelihood.classification.ClassificationLikelihood property) (fortuna.likelihood.regression.RegressionLikelihood property) (fortuna.model.model_manager.base.ModelManager property) (fortuna.model.model_manager.classification.SNGPClassificationModelManager property) (fortuna.model.model_manager.regression.RegressionModelManager property) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive property) (fortuna.output_calib_model.predictive.regression.RegressionPredictive property) (fortuna.prob_model.joint.base.Joint property) (fortuna.prob_model.predictive.classification.ClassificationPredictive property) (fortuna.prob_model.predictive.regression.RegressionPredictive property) (fortuna.prob_model.prior.base.Prior property) (fortuna.prob_model.prior.gaussian.DiagonalGaussianPrior property) (fortuna.prob_model.prior.gaussian.IsotropicGaussianPrior property) (fortuna.prob_output_layer.base.ProbOutputLayer property) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer property) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer property) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer property) root_mean_absolute_error() (in module fortuna.metric.regression) root_mean_squared_error() (in module fortuna.metric.regression) S sample() (fortuna.data.loader.DataLoader method) (fortuna.data.loader.InputsLoader method) (fortuna.data.loader.TargetsLoader method) (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.posterior.base.Posterior method) (fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_posterior.DeepEnsemblePosterior method) (fortuna.prob_model.posterior.laplace.laplace_posterior.LaplacePosterior method) (fortuna.prob_model.posterior.map.map_posterior.MAPPosterior method) (fortuna.prob_model.posterior.normalizing_flow.advi.advi_posterior.ADVIPosterior method) (fortuna.prob_model.posterior.swag.swag_posterior.SWAGPosterior method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.prior.base.Prior method) (fortuna.prob_model.prior.gaussian.DiagonalGaussianPrior method) (fortuna.prob_model.prior.gaussian.IsotropicGaussianPrior method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) sample_calibrated_outputs() (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) save_state() (fortuna.output_calib_model.base.OutputCalibModel method) (fortuna.output_calib_model.classification.OutputCalibClassifier method) (fortuna.output_calib_model.regression.OutputCalibRegressor method) (fortuna.prob_model.base.ProbModel method) (fortuna.prob_model.classification.ProbClassifier method) (fortuna.prob_model.posterior.base.Posterior method) (fortuna.prob_model.posterior.deep_ensemble.deep_ensemble_posterior.DeepEnsemblePosterior method) (fortuna.prob_model.regression.ProbRegressor method) ScalarConstantModel (class in fortuna.model.scalar_constant) ScalarHyperparameterModel (class in fortuna.model.scalar_hyper) scope (fortuna.model_editor.base.ModelEditor attribute) (fortuna.output_calibrator.classification.ClassificationTemperatureScaler attribute) (fortuna.output_calibrator.regression.RegressionTemperatureScaler attribute) score() (fortuna.conformal.regression.onedim_uncertainty.OneDimensionalUncertaintyConformalRegressor method) (fortuna.conformal.regression.quantile.QuantileConformalRegressor method) (fortuna.ood_detection.ddu.DeepDeterministicUncertaintyOODClassifier method) (fortuna.ood_detection.mahalanobis.MalahanobisOODClassifier method) score_fn() (fortuna.conformal.classification.adaptive_conformal_classifier.AdaptiveConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.AdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.adaptive_prediction.CVPlusAdaptivePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.CVPlusSimplePredictionConformalClassifier method) (fortuna.conformal.classification.simple_prediction.SimplePredictionConformalClassifier method) (fortuna.conformal.regression.batch_mvp.BatchMVPConformalRegressor method) (in module fortuna.conformal.classification.adaptive_prediction) (in module fortuna.conformal.classification.simple_prediction) seed (fortuna.model.utils.random_features.RandomFourierFeatures attribute) setup() (fortuna.model_editor.base.ModelEditor method) SGHMCPosterior (class in fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_posterior) SGHMCPosteriorApproximator (class in fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_approximator) SGHMCState (class in fortuna.prob_model.posterior.sgmcmc.sghmc.sghmc_state) SimplePredictionConformalClassifier (class in fortuna.conformal.classification.simple_prediction) size (fortuna.data.loader.DataLoader property) (fortuna.data.loader.InputsLoader property) (fortuna.data.loader.TargetsLoader property) SNGPClassificationModelManager (class in fortuna.model.model_manager.classification) SNGPPosterior (class in fortuna.prob_model.posterior.sngp.sngp_posterior) SNGPPosteriorApproximator (class in fortuna.prob_model.posterior.sngp.sngp_approximator) sow() (fortuna.model_editor.base.ModelEditor method) SpectralNormalization (class in fortuna.model.utils.spectral_norm) SpectralNormalizationConv2D (class in fortuna.model.utils.spectral_norm) split() (fortuna.data.loader.DataLoader method) (fortuna.data.loader.InputsLoader method) (fortuna.data.loader.TargetsLoader method) stage_sizes (fortuna.model.resnet.ResNet attribute) Status (in module fortuna.typing) std (fortuna.prob_model.posterior.swag.swag_state.SWAGState attribute) std() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) SWAGPosterior (class in fortuna.prob_model.posterior.swag.swag_posterior) SWAGPosteriorApproximator (class in fortuna.prob_model.posterior.swag.swag_approximator) SWAGState (class in fortuna.prob_model.posterior.swag.swag_state) T tabulate() (fortuna.model_editor.base.ModelEditor method) TargetsLoader (class in fortuna.data.loader) to_array_data() (fortuna.data.loader.DataLoader method) to_array_inputs() (fortuna.data.loader.DataLoader method) (fortuna.data.loader.InputsLoader method) to_array_targets() (fortuna.data.loader.DataLoader method) (fortuna.data.loader.TargetsLoader method) to_inputs_loader() (fortuna.data.loader.DataLoader method) to_targets_loader() (fortuna.data.loader.DataLoader method) to_transformed_data_loader() (fortuna.data.loader.DataLoader method) to_transformed_inputs_loader() (fortuna.data.loader.InputsLoader method) to_transformed_targets_loader() (fortuna.data.loader.TargetsLoader method) train() (fortuna.prob_model.base.ProbModel method) (fortuna.prob_model.classification.ProbClassifier method) (fortuna.prob_model.regression.ProbRegressor method) training_epoch_end() (fortuna.prob_model.fit_config.callback.Callback method) (fortuna.prob_model.posterior.sngp.sngp_callback.ResetCovarianceCallback method) (fortuna.training.callback.Callback method) training_epoch_start() (fortuna.prob_model.fit_config.callback.Callback method) (fortuna.prob_model.posterior.sngp.sngp_callback.ResetCovarianceCallback method) (fortuna.training.callback.Callback method) training_step_end() (fortuna.prob_model.fit_config.callback.Callback method) (fortuna.prob_model.posterior.sngp.sngp_callback.ResetCovarianceCallback method) (fortuna.training.callback.Callback method) U u_init (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) unbind() (fortuna.model_editor.base.ModelEditor method) update() (fortuna.prob_model.posterior.swag.swag_state.SWAGState method) update_error() (fortuna.conformal.classification.adaptive_conformal_classifier.AdaptiveConformalClassifier method) (fortuna.conformal.regression.adaptive_conformal_regressor.AdaptiveConformalRegressor method) update_singular_value_estimate (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) V v_init (fortuna.model.utils.spectral_norm.SpectralNormalization attribute) (fortuna.model.utils.spectral_norm.SpectralNormalizationConv2D attribute) variable() (fortuna.model_editor.base.ModelEditor method) variables (fortuna.model_editor.base.ModelEditor property) variance() (fortuna.likelihood.base.Likelihood method) (fortuna.likelihood.classification.ClassificationLikelihood method) (fortuna.likelihood.regression.RegressionLikelihood method) (fortuna.output_calib_model.predictive.classification.ClassificationPredictive method) (fortuna.output_calib_model.predictive.regression.RegressionPredictive method) (fortuna.prob_model.predictive.classification.ClassificationPredictive method) (fortuna.prob_model.predictive.regression.RegressionPredictive method) (fortuna.prob_output_layer.base.ProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationMaskedProbOutputLayer method) (fortuna.prob_output_layer.classification.ClassificationProbOutputLayer method) (fortuna.prob_output_layer.regression.RegressionProbOutputLayer method) W widen_factor (fortuna.model.wideresnet.WideResNet attribute) WideResNet (class in fortuna.model.wideresnet) WideResNet28_10 (in module fortuna.model.wideresnet) widths (fortuna.model.mlp.MLP attribute)