-
- Downloads
Add uncertainty quantification methods (#816)
* * adding changes from linter. * squashing last few PR commits into one. * PR is still 1) uncertainty (auc) methods, 2) datagen refactoring, 3) minmax for functional particle VI.
* Fix some docstrings (#833)
* adding changes:
* adding test for eval_uncertainty
* removing some unused code for packages.
* created file sl_utils with loss functions and uncertainty related functions
* use random data instead of saving/loading mnist data for eval_uncertainty test, also incorporate changes from master
* * integrated alf configurable data_loaders
* fixed docstrings according to comments
Co-authored-by:
emailweixu <emailweixu@gmail.com>
Showing
- .gitignore 4 additions, 1 deletion.gitignore
- alf/algorithms/config.py 1 addition, 1 deletionalf/algorithms/config.py
- alf/algorithms/functional_particle_vi_algorithm.py 126 additions, 32 deletionsalf/algorithms/functional_particle_vi_algorithm.py
- alf/algorithms/functional_particle_vi_algorithm_test.py 73 additions, 5 deletionsalf/algorithms/functional_particle_vi_algorithm_test.py
- alf/algorithms/generator.py 4 additions, 4 deletionsalf/algorithms/generator.py
- alf/algorithms/generator_test.py 3 additions, 1 deletionalf/algorithms/generator_test.py
- alf/algorithms/hypernetwork_algorithm.py 107 additions, 54 deletionsalf/algorithms/hypernetwork_algorithm.py
- alf/algorithms/hypernetwork_algorithm_test.py 71 additions, 10 deletionsalf/algorithms/hypernetwork_algorithm_test.py
- alf/algorithms/particle_vi_algorithm.py 94 additions, 0 deletionsalf/algorithms/particle_vi_algorithm.py
- alf/algorithms/particle_vi_algorithm_test.py 4 additions, 1 deletionalf/algorithms/particle_vi_algorithm_test.py
- alf/examples/amortized_parvi_mnist.gin 54 additions, 0 deletionsalf/examples/amortized_parvi_mnist.gin
- alf/examples/functional_parvi_mnist.gin 47 additions, 0 deletionsalf/examples/functional_parvi_mnist.gin
- alf/trainers/policy_trainer.py 8 additions, 2 deletionsalf/trainers/policy_trainer.py
- alf/trainers/policy_trainer_test.py 2 additions, 0 deletionsalf/trainers/policy_trainer_test.py
- alf/utils/datagen.py 81 additions, 106 deletionsalf/utils/datagen.py
- alf/utils/sl_utils.py 129 additions, 0 deletionsalf/utils/sl_utils.py
Loading
Please register or sign in to comment