brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch
http://probflow.readthedocs.io
MIT License
171 stars 17 forks source link

Add probabalistic kwarg to Dense, DenseRegression, and Embedding #31

Closed brendanhasz closed 3 years ago

brendanhasz commented 3 years ago

This adds a probabilistic kwarg to the Dense, DenseRegression, and Embedding modules. When probabilistic=True, Normal distributions are used for the variational posteriors, but when probabilistic=False, Deterministic distributions are used instead.

Also updates some callbacks to track walltime, adds calibration methods (calibration_curve, calibration_curve_plot, and expected_calibration_error) to ContinuousModel, and adds a Neural Linear Model example which uses these features.

Will fix https://github.com/brendanhasz/probflow/issues/18

So, all in all: