For BNNs, we use the approximation from “Bayes by Backprop” [1]}, implemented through the blitz Python package [2]}. The network has two hidden layers, with 512 neurons for the first layer, and 128 neurons for the second layer. We use a regularization coefficient of \(0.1\) .
