I’m trying to use bayesian neural networks, adding a bayesian layer to a neural networks model. Here is the code I’m using and it’s from the keras website.