├── .github ├── ISSUE_TEMPLATE │ ├── 01_BUG_REPORT.md │ ├── 02_FEATURE_REQUEST.md │ ├── 03_DOCUMENTATION.md │ ├── 04_CODEBASE_IMPROVEMENT.md │ └── config.yml ├── pull_request_template.md └── workflows │ ├── documentation.yml │ ├── lint.yml │ └── tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── NOTICE ├── README.rst ├── SECURITY.md ├── benchmarks ├── calibration │ └── temp_scaling │ │ └── breast_cancer_temp_scaling.py ├── confbayes │ ├── README.md │ ├── example_2moons_CB.py │ └── example_MNIST_CB.py ├── focal_vs_cross_entropy.ipynb ├── hallucination │ └── mmlu │ │ └── run.py ├── multivalid │ ├── breast_cancer_multicalibrate.py │ ├── mnist_top_label_multicalibrate.py │ └── two_moons_multicalibrate.py ├── tabular │ ├── analysis.py │ ├── dataset.py │ └── run.py ├── tabular_regressions.ipynb └── transformers │ ├── masked_language_modeling.py │ ├── prob_model_text_classification.py │ └── sagemaker_entrypoints │ └── prob_model_text_classification_config │ ├── default.yaml │ ├── hyperparams │ ├── cyclical_sgld_ll.yaml │ └── sghmc_ll.yaml │ ├── method │ ├── cyclical_sgld_ll.yaml │ └── sgmcmc_ll.yaml │ ├── model │ ├── bert.yaml │ └── roberta.yaml │ └── task │ ├── mnli.yaml │ └── sentiment.yaml ├── docs ├── Makefile ├── README.md ├── make.bat └── source │ ├── _static │ ├── fortuna_symbol.png │ ├── fortuna_symbol2.png │ ├── fortuna_symbol_white.png │ └── pipeline.png │ ├── conf.py │ ├── index.rst │ ├── installation.rst │ ├── landingpage.rst │ ├── license.rst │ ├── methods.rst │ ├── quickstart.rst │ ├── references │ ├── conformal.rst │ ├── data_loader.rst │ ├── likelihood.rst │ ├── metric.rst │ ├── model │ │ ├── builtin_models.rst │ │ ├── cnn.rst │ │ ├── constant.rst │ │ ├── hyper.rst │ │ ├── linen_module.rst │ │ ├── mlp.rst │ │ ├── model.rst │ │ ├── model_manager.rst │ │ ├── resnet.rst │ │ ├── scalar_constant.rst │ │ ├── scalar_hyper.rst │ │ ├── utils.rst │ │ ├── utils │ │ │ ├── random_features.rst │ │ │ └── spectral_norm.rst │ │ └── wideresnet.rst │ ├── ood_classifier.rst │ ├── output_calib_model │ │ ├── config.rst │ │ ├── output_calib_model.rst │ │ └── predictive.rst │ ├── output_calibrator.rst │ ├── plot.rst │ ├── prob_model │ │ ├── callbacks.rst │ │ ├── fit_config.rst │ │ ├── joint.rst │ │ ├── model_editor.rst │ │ ├── posterior │ │ │ ├── advi.rst │ │ │ ├── deep_ensemble.rst │ │ │ ├── laplace.rst │ │ │ ├── map.rst │ │ │ ├── posterior.rst │ │ │ ├── sgmcmc.rst │ │ │ ├── sngp.rst │ │ │ └── swag.rst │ │ ├── predictive.rst │ │ ├── prior.rst │ │ ├── prob_calib_config.rst │ │ └── prob_model.rst │ ├── prob_output_layer.rst │ ├── references.rst │ ├── sagemaker.rst │ ├── typing.rst │ └── utils.rst │ └── usage_modes │ ├── flax_models.rst │ ├── model_outputs.rst │ ├── uncertainty_estimates.rst │ └── usage_modes.rst ├── examples ├── adaptive_conformal_inference.pct.py ├── bring_in_your_own.pct.py ├── enbpi_ts_regression.pct.py ├── index.rst ├── jackknifeplus_regression.pct.py ├── mnist_classification.pct.py ├── mnist_classification_sghmc.pct.py ├── multivalid_coverage.pct.py ├── scaling_up_bayesian_inference.pct.py ├── sentiment_analysis.pct.py ├── sgmcmc_diagnostics.pct.py ├── sinusoidal_regression.pct.py ├── subnet_calibration.pct.py ├── two_moons_classification.pct.py └── two_moons_classification_ood.pct.py ├── fortuna ├── __init__.py ├── calib_model │ ├── __init__.py │ ├── base.py │ ├── calib_mixin.py │ ├── calib_model_calibrator.py │ ├── calib_state_repository.py │ ├── classification.py │ ├── config │ │ ├── __init__.py │ │ ├── base.py │ │ ├── callback.py │ │ ├── checkpointer.py │ │ ├── hyperparameters.py │ │ ├── monitor.py │ │ ├── optimizer.py │ │ └── processor.py │ ├── loss.py │ ├── predictive │ │ ├── __init__.py │ │ ├── base.py │ │ ├── classification.py │ │ └── regression.py │ ├── regression.py │ └── state.py ├── calibration │ ├── __init__.py │ ├── binary_classification │ │ ├── __init__.py │ │ └── temp_scaling │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── bias_binary_temp_scaling.py │ │ │ ├── brier_binary_temp_scaling.py │ │ │ ├── crossentropy_binary_temp_scaling.py │ │ │ └── f1_temp_scaling.py │ └── classification │ │ ├── __init__.py │ │ └── temp_scaling │ │ ├── __init__.py │ │ └── base.py ├── conformal │ ├── __init__.py │ ├── classification │ │ ├── __init__.py │ │ ├── adaptive_conformal_classifier.py │ │ ├── adaptive_prediction.py │ │ ├── base.py │ │ ├── maxcovfixprec_binary_classfication.py │ │ └── simple_prediction.py │ ├── multivalid │ │ ├── __init__.py │ │ ├── base.py │ │ ├── iterative │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── batch_mvp.py │ │ │ ├── classification │ │ │ │ ├── __init__.py │ │ │ │ ├── binary_multicalibrator.py │ │ │ │ └── top_label_multicalibrator.py │ │ │ ├── multicalibrator.py │ │ │ └── regression │ │ │ │ ├── __init__.py │ │ │ │ └── batch_mvp.py │ │ ├── mixins │ │ │ ├── __init__.py │ │ │ ├── batchmvp.py │ │ │ ├── classification │ │ │ │ ├── __init__.py │ │ │ │ ├── binary_multicalibrator.py │ │ │ │ └── top_label_multicalibrator.py │ │ │ ├── multicalibrator.py │ │ │ └── regression │ │ │ │ └── __init__.py │ │ └── one_shot │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── classification │ │ │ ├── __init__.py │ │ │ ├── binary_multicalibrator.py │ │ │ └── top_label_multicalibrator.py │ │ │ └── multicalibrator.py │ └── regression │ │ ├── __init__.py │ │ ├── adaptive_conformal_regressor.py │ │ ├── base.py │ │ ├── batch_mvp.py │ │ ├── cvplus.py │ │ ├── enbpi.py │ │ ├── jackknife_minmax.py │ │ ├── jackknifeplus.py │ │ ├── onedim_uncertainty.py │ │ └── quantile.py ├── data │ ├── __init__.py │ ├── dataset │ │ ├── __init__.py │ │ ├── data_collator.py │ │ └── huggingface_datasets.py │ └── loader │ │ ├── __init__.py │ │ ├── array_loaders.py │ │ ├── base.py │ │ ├── huggingface_loaders.py │ │ └── utils.py ├── distribution │ ├── __init__.py │ ├── base.py │ └── gaussian.py ├── docker │ ├── Dockerfile │ └── build_and_push.sh ├── hallucination │ ├── __init__.py │ ├── base.py │ ├── grouping │ │ ├── __init__.py │ │ └── clustering │ │ │ ├── __init__.py │ │ │ └── base.py │ ├── scoring │ │ └── inv_perplexity.py │ └── utils │ │ ├── __init__.py │ │ └── string.py ├── kernel_regression │ ├── __init__.py │ ├── kernels │ │ └── gaussian.py │ └── nadaraya_watson.py ├── likelihood │ ├── __init__.py │ ├── base.py │ ├── classification.py │ └── regression.py ├── loss │ ├── __init__.py │ ├── classification │ │ ├── __init__.py │ │ ├── cross_entropy.py │ │ └── focal_loss.py │ └── regression │ │ ├── __init__.py │ │ └── scaled_mse.py ├── metric │ ├── __init__.py │ ├── classification.py │ └── regression.py ├── model │ ├── __init__.py │ ├── cnn.py │ ├── constant.py │ ├── hyper.py │ ├── lenet.py │ ├── linear.py │ ├── mlp.py │ ├── model_manager │ │ ├── __init__.py │ │ ├── base.py │ │ ├── classification.py │ │ ├── name_to_model_manager.py │ │ ├── regression.py │ │ ├── state.py │ │ └── transformers │ │ │ ├── __init__.py │ │ │ └── classification.py │ ├── resnet.py │ ├── scalar_constant.py │ ├── scalar_hyper.py │ ├── utils │ │ ├── __init__.py │ │ ├── random_features.py │ │ └── spectral_norm.py │ └── wideresnet.py ├── model_editor │ ├── __init__.py │ ├── base.py │ └── probit.py ├── ood_detection │ ├── __init__.py │ ├── base.py │ ├── ddu.py │ └── mahalanobis.py ├── output_calib_model │ ├── __init__.py │ ├── base.py │ ├── classification.py │ ├── config │ │ ├── __init__.py │ │ ├── base.py │ │ ├── checkpointer.py │ │ ├── monitor.py │ │ ├── optimizer.py │ │ └── processor.py │ ├── loss.py │ ├── output_calib_mixin.py │ ├── output_calib_model_calibrator.py │ ├── output_calib_state_repository.py │ ├── predictive │ │ ├── __init__.py │ │ ├── base.py │ │ ├── classification.py │ │ └── regression.py │ ├── regression.py │ └── state.py ├── output_calibrator │ ├── __init__.py │ ├── classification.py │ ├── output_calib_manager │ │ ├── __init__.py │ │ ├── base.py │ │ └── state.py │ └── regression.py ├── plot.py ├── prob_model │ ├── __init__.py │ ├── base.py │ ├── calib_config │ │ ├── __init__.py │ │ ├── base.py │ │ ├── checkpointer.py │ │ ├── monitor.py │ │ ├── optimizer.py │ │ └── processor.py │ ├── classification.py │ ├── fit_config │ │ ├── __init__.py │ │ ├── base.py │ │ ├── callback.py │ │ ├── checkpointer.py │ │ ├── hyperparameters.py │ │ ├── monitor.py │ │ ├── optimizer.py │ │ └── processor.py │ ├── joint │ │ ├── __init__.py │ │ ├── base.py │ │ └── state.py │ ├── posterior │ │ ├── __init__.py │ │ ├── base.py │ │ ├── deep_ensemble │ │ │ ├── __init__.py │ │ │ ├── deep_ensemble_approximator.py │ │ │ ├── deep_ensemble_posterior.py │ │ │ └── deep_ensemble_state.py │ │ ├── laplace │ │ │ ├── __init__.py │ │ │ ├── laplace_approximator.py │ │ │ ├── laplace_posterior.py │ │ │ └── laplace_state.py │ │ ├── map │ │ │ ├── __init__.py │ │ │ ├── map_approximator.py │ │ │ ├── map_posterior.py │ │ │ ├── map_state.py │ │ │ └── map_trainer.py │ │ ├── name_to_posterior_state.py │ │ ├── normalizing_flow │ │ │ ├── __init__.py │ │ │ ├── advi │ │ │ │ ├── __init__.py │ │ │ │ ├── advi_approximator.py │ │ │ │ ├── advi_architecture.py │ │ │ │ ├── advi_posterior.py │ │ │ │ ├── advi_state.py │ │ │ │ └── advi_trainer.py │ │ │ ├── normalizing_flow_state.py │ │ │ └── normalizing_flow_trainer.py │ │ ├── posterior_approximations.py │ │ ├── posterior_mixin.py │ │ ├── posterior_multi_state_repository.py │ │ ├── posterior_state_repository.py │ │ ├── posterior_trainer.py │ │ ├── run_preliminary_map.py │ │ ├── sgmcmc │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── cyclical_sgld │ │ │ │ ├── __init__.py │ │ │ │ ├── cyclical_sgld_approximator.py │ │ │ │ ├── cyclical_sgld_callback.py │ │ │ │ ├── cyclical_sgld_integrator.py │ │ │ │ ├── cyclical_sgld_posterior.py │ │ │ │ └── cyclical_sgld_state.py │ │ │ ├── sghmc │ │ │ │ ├── __init__.py │ │ │ │ ├── sghmc_approximator.py │ │ │ │ ├── sghmc_callback.py │ │ │ │ ├── sghmc_integrator.py │ │ │ │ ├── sghmc_posterior.py │ │ │ │ └── sghmc_state.py │ │ │ ├── sgmcmc_diagnostic.py │ │ │ ├── sgmcmc_posterior.py │ │ │ ├── sgmcmc_posterior_state_repository.py │ │ │ ├── sgmcmc_preconditioner.py │ │ │ ├── sgmcmc_sampling_callback.py │ │ │ └── sgmcmc_step_schedule.py │ │ ├── sngp │ │ │ ├── __init__.py │ │ │ ├── sngp_approximator.py │ │ │ ├── sngp_callback.py │ │ │ ├── sngp_posterior.py │ │ │ └── transformers │ │ │ │ ├── __init__.py │ │ │ │ ├── auto_factory.py │ │ │ │ ├── modeling_flax_auto.py │ │ │ │ └── models │ │ │ │ ├── __init__.py │ │ │ │ ├── modeling_flax_bert.py │ │ │ │ ├── modeling_flax_distilbert.py │ │ │ │ └── modeling_flax_roberta.py │ │ ├── state.py │ │ └── swag │ │ │ ├── __init__.py │ │ │ ├── swag_approximator.py │ │ │ ├── swag_posterior.py │ │ │ ├── swag_state.py │ │ │ └── swag_trainer.py │ ├── predictive │ │ ├── __init__.py │ │ ├── base.py │ │ ├── classification.py │ │ └── regression.py │ ├── prior │ │ ├── __init__.py │ │ ├── base.py │ │ └── gaussian.py │ ├── prob_model_calibrator.py │ ├── regression.py │ └── state.py ├── prob_output_layer │ ├── __init__.py │ ├── base.py │ ├── classification.py │ └── regression.py ├── sagemaker │ ├── __init__.py │ ├── base.py │ └── utils.py ├── training │ ├── __init__.py │ ├── callback.py │ ├── mixin.py │ ├── name_to_train_state.py │ ├── output_calibrator.py │ ├── train_state.py │ ├── train_state_repository.py │ └── trainer.py ├── typing.py └── utils │ ├── __init__.py │ ├── builtins.py │ ├── data.py │ ├── device.py │ ├── freeze.py │ ├── grad.py │ ├── nested_dicts.py │ ├── optimizer.py │ ├── probit.py │ ├── random.py │ ├── strings.py │ └── training.py ├── poetry.lock ├── pyproject.toml └── tests ├── __init__.py ├── fortuna ├── __init__.py ├── calib_model │ ├── test_calib_model.py │ └── test_output_calib_model.py ├── hallucination │ ├── grouping.py │ └── scoring.py ├── prob_model │ ├── __init__.py │ ├── test_diagnostic.py │ ├── test_joint.py │ ├── test_likelihood.py │ ├── test_preconditioner.py │ ├── test_prob_model.py │ ├── test_step_schedule.py │ └── test_train.py ├── test_builtins.py ├── test_conformal_methods.py ├── test_data.py ├── test_kernel_regression.py ├── test_metric.py ├── test_mixin.py ├── test_model.py ├── test_output_maker.py ├── test_plot.py ├── test_predictive.py ├── test_prior.py ├── test_prob_output_layer.py ├── test_state.py ├── test_temp_scaling.py ├── test_trainer.py └── utils │ └── freeze.py ├── make_data.py └── make_model.py /.github/ISSUE_TEMPLATE/01_BUG_REPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/ISSUE_TEMPLATE/01_BUG_REPORT.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/02_FEATURE_REQUEST.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/ISSUE_TEMPLATE/02_FEATURE_REQUEST.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/03_DOCUMENTATION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/ISSUE_TEMPLATE/03_DOCUMENTATION.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/04_CODEBASE_IMPROVEMENT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/ISSUE_TEMPLATE/04_CODEBASE_IMPROVEMENT.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.github/workflows/documentation.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/workflows/documentation.yml -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.github/workflows/tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- 1 | Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/README.rst -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/SECURITY.md -------------------------------------------------------------------------------- /benchmarks/calibration/temp_scaling/breast_cancer_temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/calibration/temp_scaling/breast_cancer_temp_scaling.py -------------------------------------------------------------------------------- /benchmarks/confbayes/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/confbayes/README.md -------------------------------------------------------------------------------- /benchmarks/confbayes/example_2moons_CB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/confbayes/example_2moons_CB.py -------------------------------------------------------------------------------- /benchmarks/confbayes/example_MNIST_CB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/confbayes/example_MNIST_CB.py -------------------------------------------------------------------------------- /benchmarks/focal_vs_cross_entropy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/focal_vs_cross_entropy.ipynb -------------------------------------------------------------------------------- /benchmarks/hallucination/mmlu/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/hallucination/mmlu/run.py -------------------------------------------------------------------------------- /benchmarks/multivalid/breast_cancer_multicalibrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/multivalid/breast_cancer_multicalibrate.py -------------------------------------------------------------------------------- /benchmarks/multivalid/mnist_top_label_multicalibrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/multivalid/mnist_top_label_multicalibrate.py -------------------------------------------------------------------------------- /benchmarks/multivalid/two_moons_multicalibrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/multivalid/two_moons_multicalibrate.py -------------------------------------------------------------------------------- /benchmarks/tabular/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/tabular/analysis.py -------------------------------------------------------------------------------- /benchmarks/tabular/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/tabular/dataset.py -------------------------------------------------------------------------------- /benchmarks/tabular/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/tabular/run.py -------------------------------------------------------------------------------- /benchmarks/tabular_regressions.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/tabular_regressions.ipynb -------------------------------------------------------------------------------- /benchmarks/transformers/masked_language_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/masked_language_modeling.py -------------------------------------------------------------------------------- /benchmarks/transformers/prob_model_text_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/prob_model_text_classification.py -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/default.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/hyperparams/cyclical_sgld_ll.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/hyperparams/cyclical_sgld_ll.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/hyperparams/sghmc_ll.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/hyperparams/sghmc_ll.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/method/cyclical_sgld_ll.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/method/cyclical_sgld_ll.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/method/sgmcmc_ll.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/method/sgmcmc_ll.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/model/bert.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/model/bert.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/model/roberta.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/model/roberta.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/task/mnli.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/task/mnli.yaml -------------------------------------------------------------------------------- /benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/task/sentiment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/benchmarks/transformers/sagemaker_entrypoints/prob_model_text_classification_config/task/sentiment.yaml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/source/_static/fortuna_symbol.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/_static/fortuna_symbol.png -------------------------------------------------------------------------------- /docs/source/_static/fortuna_symbol2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/_static/fortuna_symbol2.png -------------------------------------------------------------------------------- /docs/source/_static/fortuna_symbol_white.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/_static/fortuna_symbol_white.png -------------------------------------------------------------------------------- /docs/source/_static/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/_static/pipeline.png -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/installation.rst -------------------------------------------------------------------------------- /docs/source/landingpage.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/landingpage.rst -------------------------------------------------------------------------------- /docs/source/license.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/license.rst -------------------------------------------------------------------------------- /docs/source/methods.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/methods.rst -------------------------------------------------------------------------------- /docs/source/quickstart.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/quickstart.rst -------------------------------------------------------------------------------- /docs/source/references/conformal.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/conformal.rst -------------------------------------------------------------------------------- /docs/source/references/data_loader.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/data_loader.rst -------------------------------------------------------------------------------- /docs/source/references/likelihood.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/likelihood.rst -------------------------------------------------------------------------------- /docs/source/references/metric.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/metric.rst -------------------------------------------------------------------------------- /docs/source/references/model/builtin_models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/builtin_models.rst -------------------------------------------------------------------------------- /docs/source/references/model/cnn.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/cnn.rst -------------------------------------------------------------------------------- /docs/source/references/model/constant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/constant.rst -------------------------------------------------------------------------------- /docs/source/references/model/hyper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/hyper.rst -------------------------------------------------------------------------------- /docs/source/references/model/linen_module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/linen_module.rst -------------------------------------------------------------------------------- /docs/source/references/model/mlp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/mlp.rst -------------------------------------------------------------------------------- /docs/source/references/model/model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/model.rst -------------------------------------------------------------------------------- /docs/source/references/model/model_manager.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/model_manager.rst -------------------------------------------------------------------------------- /docs/source/references/model/resnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/resnet.rst -------------------------------------------------------------------------------- /docs/source/references/model/scalar_constant.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/scalar_constant.rst -------------------------------------------------------------------------------- /docs/source/references/model/scalar_hyper.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/scalar_hyper.rst -------------------------------------------------------------------------------- /docs/source/references/model/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/utils.rst -------------------------------------------------------------------------------- /docs/source/references/model/utils/random_features.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/utils/random_features.rst -------------------------------------------------------------------------------- /docs/source/references/model/utils/spectral_norm.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/utils/spectral_norm.rst -------------------------------------------------------------------------------- /docs/source/references/model/wideresnet.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/model/wideresnet.rst -------------------------------------------------------------------------------- /docs/source/references/ood_classifier.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/ood_classifier.rst -------------------------------------------------------------------------------- /docs/source/references/output_calib_model/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/output_calib_model/config.rst -------------------------------------------------------------------------------- /docs/source/references/output_calib_model/output_calib_model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/output_calib_model/output_calib_model.rst -------------------------------------------------------------------------------- /docs/source/references/output_calib_model/predictive.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/output_calib_model/predictive.rst -------------------------------------------------------------------------------- /docs/source/references/output_calibrator.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/output_calibrator.rst -------------------------------------------------------------------------------- /docs/source/references/plot.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/plot.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/callbacks.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/callbacks.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/fit_config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/fit_config.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/joint.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/joint.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/model_editor.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/model_editor.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/advi.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/advi.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/deep_ensemble.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/deep_ensemble.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/laplace.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/laplace.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/map.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/map.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/posterior.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/posterior.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/sgmcmc.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/sgmcmc.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/sngp.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/sngp.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/posterior/swag.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/posterior/swag.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/predictive.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/predictive.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/prior.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/prior.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/prob_calib_config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/prob_calib_config.rst -------------------------------------------------------------------------------- /docs/source/references/prob_model/prob_model.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_model/prob_model.rst -------------------------------------------------------------------------------- /docs/source/references/prob_output_layer.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/prob_output_layer.rst -------------------------------------------------------------------------------- /docs/source/references/references.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/references.rst -------------------------------------------------------------------------------- /docs/source/references/sagemaker.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/sagemaker.rst -------------------------------------------------------------------------------- /docs/source/references/typing.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/typing.rst -------------------------------------------------------------------------------- /docs/source/references/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/references/utils.rst -------------------------------------------------------------------------------- /docs/source/usage_modes/flax_models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/usage_modes/flax_models.rst -------------------------------------------------------------------------------- /docs/source/usage_modes/model_outputs.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/usage_modes/model_outputs.rst -------------------------------------------------------------------------------- /docs/source/usage_modes/uncertainty_estimates.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/usage_modes/uncertainty_estimates.rst -------------------------------------------------------------------------------- /docs/source/usage_modes/usage_modes.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/docs/source/usage_modes/usage_modes.rst -------------------------------------------------------------------------------- /examples/adaptive_conformal_inference.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/adaptive_conformal_inference.pct.py -------------------------------------------------------------------------------- /examples/bring_in_your_own.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/bring_in_your_own.pct.py -------------------------------------------------------------------------------- /examples/enbpi_ts_regression.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/enbpi_ts_regression.pct.py -------------------------------------------------------------------------------- /examples/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/index.rst -------------------------------------------------------------------------------- /examples/jackknifeplus_regression.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/jackknifeplus_regression.pct.py -------------------------------------------------------------------------------- /examples/mnist_classification.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/mnist_classification.pct.py -------------------------------------------------------------------------------- /examples/mnist_classification_sghmc.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/mnist_classification_sghmc.pct.py -------------------------------------------------------------------------------- /examples/multivalid_coverage.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/multivalid_coverage.pct.py -------------------------------------------------------------------------------- /examples/scaling_up_bayesian_inference.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/scaling_up_bayesian_inference.pct.py -------------------------------------------------------------------------------- /examples/sentiment_analysis.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/sentiment_analysis.pct.py -------------------------------------------------------------------------------- /examples/sgmcmc_diagnostics.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/sgmcmc_diagnostics.pct.py -------------------------------------------------------------------------------- /examples/sinusoidal_regression.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/sinusoidal_regression.pct.py -------------------------------------------------------------------------------- /examples/subnet_calibration.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/subnet_calibration.pct.py -------------------------------------------------------------------------------- /examples/two_moons_classification.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/two_moons_classification.pct.py -------------------------------------------------------------------------------- /examples/two_moons_classification_ood.pct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/examples/two_moons_classification_ood.pct.py -------------------------------------------------------------------------------- /fortuna/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/__init__.py -------------------------------------------------------------------------------- /fortuna/calib_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/__init__.py -------------------------------------------------------------------------------- /fortuna/calib_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/base.py -------------------------------------------------------------------------------- /fortuna/calib_model/calib_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/calib_mixin.py -------------------------------------------------------------------------------- /fortuna/calib_model/calib_model_calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/calib_model_calibrator.py -------------------------------------------------------------------------------- /fortuna/calib_model/calib_state_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/calib_state_repository.py -------------------------------------------------------------------------------- /fortuna/calib_model/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/classification.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/calib_model/config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/base.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/callback.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/checkpointer.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/hyperparameters.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/monitor.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/optimizer.py -------------------------------------------------------------------------------- /fortuna/calib_model/config/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/config/processor.py -------------------------------------------------------------------------------- /fortuna/calib_model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/loss.py -------------------------------------------------------------------------------- /fortuna/calib_model/predictive/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/calib_model/predictive/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/predictive/base.py -------------------------------------------------------------------------------- /fortuna/calib_model/predictive/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/predictive/classification.py -------------------------------------------------------------------------------- /fortuna/calib_model/predictive/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/predictive/regression.py -------------------------------------------------------------------------------- /fortuna/calib_model/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/regression.py -------------------------------------------------------------------------------- /fortuna/calib_model/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calib_model/state.py -------------------------------------------------------------------------------- /fortuna/calibration/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/__init__.py -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/temp_scaling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/temp_scaling/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/binary_classification/temp_scaling/base.py -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/temp_scaling/bias_binary_temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/binary_classification/temp_scaling/bias_binary_temp_scaling.py -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/temp_scaling/brier_binary_temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/binary_classification/temp_scaling/brier_binary_temp_scaling.py -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/temp_scaling/crossentropy_binary_temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/binary_classification/temp_scaling/crossentropy_binary_temp_scaling.py -------------------------------------------------------------------------------- /fortuna/calibration/binary_classification/temp_scaling/f1_temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/binary_classification/temp_scaling/f1_temp_scaling.py -------------------------------------------------------------------------------- /fortuna/calibration/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/calibration/classification/temp_scaling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/calibration/classification/temp_scaling/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/calibration/classification/temp_scaling/base.py -------------------------------------------------------------------------------- /fortuna/conformal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/__init__.py -------------------------------------------------------------------------------- /fortuna/conformal/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/classification/adaptive_conformal_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/classification/adaptive_conformal_classifier.py -------------------------------------------------------------------------------- /fortuna/conformal/classification/adaptive_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/classification/adaptive_prediction.py -------------------------------------------------------------------------------- /fortuna/conformal/classification/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/classification/base.py -------------------------------------------------------------------------------- /fortuna/conformal/classification/maxcovfixprec_binary_classfication.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/classification/maxcovfixprec_binary_classfication.py -------------------------------------------------------------------------------- /fortuna/conformal/classification/simple_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/classification/simple_prediction.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/base.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/iterative/base.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/batch_mvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/iterative/batch_mvp.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/classification/binary_multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/iterative/classification/binary_multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/classification/top_label_multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/iterative/classification/top_label_multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/iterative/multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/regression/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/iterative/regression/batch_mvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/iterative/regression/batch_mvp.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/batchmvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/mixins/batchmvp.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/classification/binary_multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/mixins/classification/binary_multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/classification/top_label_multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/mixins/classification/top_label_multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/mixins/multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/mixins/regression/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/one_shot/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/one_shot/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/one_shot/base.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/one_shot/classification/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/one_shot/classification/binary_multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/one_shot/classification/binary_multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/one_shot/classification/top_label_multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/one_shot/classification/top_label_multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/multivalid/one_shot/multicalibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/multivalid/one_shot/multicalibrator.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/conformal/regression/adaptive_conformal_regressor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/adaptive_conformal_regressor.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/base.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/batch_mvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/batch_mvp.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/cvplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/cvplus.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/enbpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/enbpi.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/jackknife_minmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/jackknife_minmax.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/jackknifeplus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/jackknifeplus.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/onedim_uncertainty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/onedim_uncertainty.py -------------------------------------------------------------------------------- /fortuna/conformal/regression/quantile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/conformal/regression/quantile.py -------------------------------------------------------------------------------- /fortuna/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/__init__.py -------------------------------------------------------------------------------- /fortuna/data/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/data/dataset/data_collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/dataset/data_collator.py -------------------------------------------------------------------------------- /fortuna/data/dataset/huggingface_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/dataset/huggingface_datasets.py -------------------------------------------------------------------------------- /fortuna/data/loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/loader/__init__.py -------------------------------------------------------------------------------- /fortuna/data/loader/array_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/loader/array_loaders.py -------------------------------------------------------------------------------- /fortuna/data/loader/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/loader/base.py -------------------------------------------------------------------------------- /fortuna/data/loader/huggingface_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/loader/huggingface_loaders.py -------------------------------------------------------------------------------- /fortuna/data/loader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/data/loader/utils.py -------------------------------------------------------------------------------- /fortuna/distribution/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/distribution/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/distribution/base.py -------------------------------------------------------------------------------- /fortuna/distribution/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/distribution/gaussian.py -------------------------------------------------------------------------------- /fortuna/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/docker/Dockerfile -------------------------------------------------------------------------------- /fortuna/docker/build_and_push.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/docker/build_and_push.sh -------------------------------------------------------------------------------- /fortuna/hallucination/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/hallucination/__init__.py -------------------------------------------------------------------------------- /fortuna/hallucination/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/hallucination/base.py -------------------------------------------------------------------------------- /fortuna/hallucination/grouping/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/hallucination/grouping/clustering/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/hallucination/grouping/clustering/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/hallucination/grouping/clustering/base.py -------------------------------------------------------------------------------- /fortuna/hallucination/scoring/inv_perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/hallucination/scoring/inv_perplexity.py -------------------------------------------------------------------------------- /fortuna/hallucination/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/hallucination/utils/__init__.py -------------------------------------------------------------------------------- /fortuna/hallucination/utils/string.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/hallucination/utils/string.py -------------------------------------------------------------------------------- /fortuna/kernel_regression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/kernel_regression/__init__.py -------------------------------------------------------------------------------- /fortuna/kernel_regression/kernels/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/kernel_regression/kernels/gaussian.py -------------------------------------------------------------------------------- /fortuna/kernel_regression/nadaraya_watson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/kernel_regression/nadaraya_watson.py -------------------------------------------------------------------------------- /fortuna/likelihood/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/likelihood/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/likelihood/base.py -------------------------------------------------------------------------------- /fortuna/likelihood/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/likelihood/classification.py -------------------------------------------------------------------------------- /fortuna/likelihood/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/likelihood/regression.py -------------------------------------------------------------------------------- /fortuna/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/loss/classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/loss/classification/__init__.py -------------------------------------------------------------------------------- /fortuna/loss/classification/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/loss/classification/cross_entropy.py -------------------------------------------------------------------------------- /fortuna/loss/classification/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/loss/classification/focal_loss.py -------------------------------------------------------------------------------- /fortuna/loss/regression/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/loss/regression/__init__.py -------------------------------------------------------------------------------- /fortuna/loss/regression/scaled_mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/loss/regression/scaled_mse.py -------------------------------------------------------------------------------- /fortuna/metric/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/metric/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/metric/classification.py -------------------------------------------------------------------------------- /fortuna/metric/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/metric/regression.py -------------------------------------------------------------------------------- /fortuna/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/__init__.py -------------------------------------------------------------------------------- /fortuna/model/cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/cnn.py -------------------------------------------------------------------------------- /fortuna/model/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/constant.py -------------------------------------------------------------------------------- /fortuna/model/hyper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/hyper.py -------------------------------------------------------------------------------- /fortuna/model/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/lenet.py -------------------------------------------------------------------------------- /fortuna/model/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/linear.py -------------------------------------------------------------------------------- /fortuna/model/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/mlp.py -------------------------------------------------------------------------------- /fortuna/model/model_manager/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/model/model_manager/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/model_manager/base.py -------------------------------------------------------------------------------- /fortuna/model/model_manager/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/model_manager/classification.py -------------------------------------------------------------------------------- /fortuna/model/model_manager/name_to_model_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/model_manager/name_to_model_manager.py -------------------------------------------------------------------------------- /fortuna/model/model_manager/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/model_manager/regression.py -------------------------------------------------------------------------------- /fortuna/model/model_manager/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/model_manager/state.py -------------------------------------------------------------------------------- /fortuna/model/model_manager/transformers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/model/model_manager/transformers/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/model_manager/transformers/classification.py -------------------------------------------------------------------------------- /fortuna/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/resnet.py -------------------------------------------------------------------------------- /fortuna/model/scalar_constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/scalar_constant.py -------------------------------------------------------------------------------- /fortuna/model/scalar_hyper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/scalar_hyper.py -------------------------------------------------------------------------------- /fortuna/model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/model/utils/random_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/utils/random_features.py -------------------------------------------------------------------------------- /fortuna/model/utils/spectral_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/utils/spectral_norm.py -------------------------------------------------------------------------------- /fortuna/model/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model/wideresnet.py -------------------------------------------------------------------------------- /fortuna/model_editor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model_editor/__init__.py -------------------------------------------------------------------------------- /fortuna/model_editor/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model_editor/base.py -------------------------------------------------------------------------------- /fortuna/model_editor/probit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/model_editor/probit.py -------------------------------------------------------------------------------- /fortuna/ood_detection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/ood_detection/__init__.py -------------------------------------------------------------------------------- /fortuna/ood_detection/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/ood_detection/base.py -------------------------------------------------------------------------------- /fortuna/ood_detection/ddu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/ood_detection/ddu.py -------------------------------------------------------------------------------- /fortuna/ood_detection/mahalanobis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/ood_detection/mahalanobis.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/__init__.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/base.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/classification.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/output_calib_model/config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/config/base.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/config/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/config/checkpointer.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/config/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/config/monitor.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/config/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/config/optimizer.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/config/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/config/processor.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/loss.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/output_calib_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/output_calib_mixin.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/output_calib_model_calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/output_calib_model_calibrator.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/output_calib_state_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/output_calib_state_repository.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/predictive/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/output_calib_model/predictive/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/predictive/base.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/predictive/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/predictive/classification.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/predictive/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/predictive/regression.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/regression.py -------------------------------------------------------------------------------- /fortuna/output_calib_model/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calib_model/state.py -------------------------------------------------------------------------------- /fortuna/output_calibrator/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/output_calibrator/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calibrator/classification.py -------------------------------------------------------------------------------- /fortuna/output_calibrator/output_calib_manager/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/output_calibrator/output_calib_manager/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calibrator/output_calib_manager/base.py -------------------------------------------------------------------------------- /fortuna/output_calibrator/output_calib_manager/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calibrator/output_calib_manager/state.py -------------------------------------------------------------------------------- /fortuna/output_calibrator/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/output_calibrator/regression.py -------------------------------------------------------------------------------- /fortuna/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/plot.py -------------------------------------------------------------------------------- /fortuna/prob_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/__init__.py -------------------------------------------------------------------------------- /fortuna/prob_model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/calib_config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/calib_config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/calib_config/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/calib_config/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/calib_config/checkpointer.py -------------------------------------------------------------------------------- /fortuna/prob_model/calib_config/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/calib_config/monitor.py -------------------------------------------------------------------------------- /fortuna/prob_model/calib_config/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/calib_config/optimizer.py -------------------------------------------------------------------------------- /fortuna/prob_model/calib_config/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/calib_config/processor.py -------------------------------------------------------------------------------- /fortuna/prob_model/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/classification.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/callback.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/checkpointer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/checkpointer.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/hyperparameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/hyperparameters.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/monitor.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/optimizer.py -------------------------------------------------------------------------------- /fortuna/prob_model/fit_config/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/fit_config/processor.py -------------------------------------------------------------------------------- /fortuna/prob_model/joint/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/joint/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/joint/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/joint/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/joint/state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/deep_ensemble/__init__.py: -------------------------------------------------------------------------------- 1 | DEEP_ENSEMBLE_NAME = "deep_ensemble" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/deep_ensemble/deep_ensemble_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/deep_ensemble/deep_ensemble_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/deep_ensemble/deep_ensemble_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/deep_ensemble/deep_ensemble_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/deep_ensemble/deep_ensemble_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/deep_ensemble/deep_ensemble_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/laplace/__init__.py: -------------------------------------------------------------------------------- 1 | LAPLACE_NAME = "laplace" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/laplace/laplace_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/laplace/laplace_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/laplace/laplace_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/laplace/laplace_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/laplace/laplace_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/laplace/laplace_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/map/__init__.py: -------------------------------------------------------------------------------- 1 | MAP_NAME = "map" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/map/map_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/map/map_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/map/map_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/map/map_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/map/map_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/map/map_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/map/map_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/map/map_trainer.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/name_to_posterior_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/name_to_posterior_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/advi/__init__.py: -------------------------------------------------------------------------------- 1 | ADVI_NAME = "advi" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/advi/advi_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/advi/advi_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/advi/advi_architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/advi/advi_architecture.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/advi/advi_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/advi/advi_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/advi/advi_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/advi/advi_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/advi/advi_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/advi/advi_trainer.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/normalizing_flow_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/normalizing_flow_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/normalizing_flow/normalizing_flow_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/normalizing_flow/normalizing_flow_trainer.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/posterior_approximations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/posterior_approximations.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/posterior_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/posterior_mixin.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/posterior_multi_state_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/posterior_multi_state_repository.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/posterior_state_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/posterior_state_repository.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/posterior_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/posterior_trainer.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/run_preliminary_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/run_preliminary_map.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/__init__.py: -------------------------------------------------------------------------------- 1 | CYCLICAL_SGLD_NAME = "cyclical_sgld" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_callback.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_integrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_integrator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/cyclical_sgld/cyclical_sgld_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sghmc/__init__.py: -------------------------------------------------------------------------------- 1 | SGHMC_NAME = "sghmc" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_callback.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_integrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_integrator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sghmc/sghmc_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sgmcmc_diagnostic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sgmcmc_diagnostic.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sgmcmc_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sgmcmc_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sgmcmc_posterior_state_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sgmcmc_posterior_state_repository.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sgmcmc_preconditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sgmcmc_preconditioner.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sgmcmc_sampling_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sgmcmc_sampling_callback.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sgmcmc/sgmcmc_step_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sgmcmc/sgmcmc_step_schedule.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/__init__.py: -------------------------------------------------------------------------------- 1 | SNGP_NAME = "sngp" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/sngp_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/sngp_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/sngp_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/sngp_callback.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/sngp_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/sngp_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/transformers/__init__.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/auto_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/transformers/auto_factory.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/modeling_flax_auto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/transformers/modeling_flax_auto.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/models/modeling_flax_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/transformers/models/modeling_flax_bert.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/models/modeling_flax_distilbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/transformers/models/modeling_flax_distilbert.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/sngp/transformers/models/modeling_flax_roberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/sngp/transformers/models/modeling_flax_roberta.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/swag/__init__.py: -------------------------------------------------------------------------------- 1 | SWAG_NAME = "swag" 2 | -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/swag/swag_approximator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/swag/swag_approximator.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/swag/swag_posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/swag/swag_posterior.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/swag/swag_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/swag/swag_state.py -------------------------------------------------------------------------------- /fortuna/prob_model/posterior/swag/swag_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/posterior/swag/swag_trainer.py -------------------------------------------------------------------------------- /fortuna/prob_model/predictive/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_model/predictive/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/predictive/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/predictive/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/predictive/classification.py -------------------------------------------------------------------------------- /fortuna/prob_model/predictive/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/predictive/regression.py -------------------------------------------------------------------------------- /fortuna/prob_model/prior/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/prior/__init__.py -------------------------------------------------------------------------------- /fortuna/prob_model/prior/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/prior/base.py -------------------------------------------------------------------------------- /fortuna/prob_model/prior/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/prior/gaussian.py -------------------------------------------------------------------------------- /fortuna/prob_model/prob_model_calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/prob_model_calibrator.py -------------------------------------------------------------------------------- /fortuna/prob_model/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/regression.py -------------------------------------------------------------------------------- /fortuna/prob_model/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_model/state.py -------------------------------------------------------------------------------- /fortuna/prob_output_layer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/prob_output_layer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_output_layer/base.py -------------------------------------------------------------------------------- /fortuna/prob_output_layer/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_output_layer/classification.py -------------------------------------------------------------------------------- /fortuna/prob_output_layer/regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/prob_output_layer/regression.py -------------------------------------------------------------------------------- /fortuna/sagemaker/__init__.py: -------------------------------------------------------------------------------- 1 | from fortuna.sagemaker.base import run_training_job 2 | -------------------------------------------------------------------------------- /fortuna/sagemaker/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/sagemaker/base.py -------------------------------------------------------------------------------- /fortuna/sagemaker/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/sagemaker/utils.py -------------------------------------------------------------------------------- /fortuna/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/training/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/callback.py -------------------------------------------------------------------------------- /fortuna/training/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/mixin.py -------------------------------------------------------------------------------- /fortuna/training/name_to_train_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/name_to_train_state.py -------------------------------------------------------------------------------- /fortuna/training/output_calibrator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/output_calibrator.py -------------------------------------------------------------------------------- /fortuna/training/train_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/train_state.py -------------------------------------------------------------------------------- /fortuna/training/train_state_repository.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/train_state_repository.py -------------------------------------------------------------------------------- /fortuna/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/training/trainer.py -------------------------------------------------------------------------------- /fortuna/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/typing.py -------------------------------------------------------------------------------- /fortuna/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /fortuna/utils/builtins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/builtins.py -------------------------------------------------------------------------------- /fortuna/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/data.py -------------------------------------------------------------------------------- /fortuna/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/device.py -------------------------------------------------------------------------------- /fortuna/utils/freeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/freeze.py -------------------------------------------------------------------------------- /fortuna/utils/grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/grad.py -------------------------------------------------------------------------------- /fortuna/utils/nested_dicts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/nested_dicts.py -------------------------------------------------------------------------------- /fortuna/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/optimizer.py -------------------------------------------------------------------------------- /fortuna/utils/probit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/probit.py -------------------------------------------------------------------------------- /fortuna/utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/random.py -------------------------------------------------------------------------------- /fortuna/utils/strings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/strings.py -------------------------------------------------------------------------------- /fortuna/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/fortuna/utils/training.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fortuna/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fortuna/calib_model/test_calib_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/calib_model/test_calib_model.py -------------------------------------------------------------------------------- /tests/fortuna/calib_model/test_output_calib_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/calib_model/test_output_calib_model.py -------------------------------------------------------------------------------- /tests/fortuna/hallucination/grouping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/hallucination/grouping.py -------------------------------------------------------------------------------- /tests/fortuna/hallucination/scoring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/hallucination/scoring.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_diagnostic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_diagnostic.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_joint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_joint.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_likelihood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_likelihood.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_preconditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_preconditioner.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_prob_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_prob_model.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_step_schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_step_schedule.py -------------------------------------------------------------------------------- /tests/fortuna/prob_model/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/prob_model/test_train.py -------------------------------------------------------------------------------- /tests/fortuna/test_builtins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_builtins.py -------------------------------------------------------------------------------- /tests/fortuna/test_conformal_methods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_conformal_methods.py -------------------------------------------------------------------------------- /tests/fortuna/test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_data.py -------------------------------------------------------------------------------- /tests/fortuna/test_kernel_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_kernel_regression.py -------------------------------------------------------------------------------- /tests/fortuna/test_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_metric.py -------------------------------------------------------------------------------- /tests/fortuna/test_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_mixin.py -------------------------------------------------------------------------------- /tests/fortuna/test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_model.py -------------------------------------------------------------------------------- /tests/fortuna/test_output_maker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_output_maker.py -------------------------------------------------------------------------------- /tests/fortuna/test_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_plot.py -------------------------------------------------------------------------------- /tests/fortuna/test_predictive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_predictive.py -------------------------------------------------------------------------------- /tests/fortuna/test_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_prior.py -------------------------------------------------------------------------------- /tests/fortuna/test_prob_output_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_prob_output_layer.py -------------------------------------------------------------------------------- /tests/fortuna/test_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_state.py -------------------------------------------------------------------------------- /tests/fortuna/test_temp_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_temp_scaling.py -------------------------------------------------------------------------------- /tests/fortuna/test_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/test_trainer.py -------------------------------------------------------------------------------- /tests/fortuna/utils/freeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/fortuna/utils/freeze.py -------------------------------------------------------------------------------- /tests/make_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/make_data.py -------------------------------------------------------------------------------- /tests/make_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/awslabs/fortuna/HEAD/tests/make_model.py --------------------------------------------------------------------------------