Clean up BayesianDenseLayer
Opened this issue · 0 comments
st-- commented
It's currently a bit confusing. What is its relationship to tfp.layers.DenseVariational, what are its advantages? If we want to keep it, it needs an overhaul of its API documentation (and demonstrating its use in a notebook). We might then also want to subclass DenseVariational or turn it into a DistributionLambda of our own.