Predicting Values with Bayesian Neural Network
I want to use a Bayesian Neural Network for a regression task. To do that I converted a BNN from this paper to Python 3. The provided training script runs and I receive a pickle file, which I want to use to predict a value for my regression. Event though the loss for the training doesn't really go down, but this seems also be the case for the models used in the paper. The input of my network is a vector with 59 features, where one feature is my target variable. It's pretty similar to the boston housing dataset. The MLP inside the BNN returns two vales: $\mu$ and $\sigma$ . If I understand BNNs correctly $\mu$ and $\sigma$ refer to the mean and standard deviation of my input and I should use $\mu$ to predict my values. But when I use input vector I always receive a value that is roughly around the mean of my target variable. So lets say the mean is 62. When I insert a vector from my test set, that should return the target variable 142, I will get 61. When I do the same with a vector, that should return 21, I will get 63. So my predicted values are always around the mean of the target variable. What do I have to do, to get real predictions from my input data? I never worked with BNNs before and it's quite hard to find any resources on how I should model my network. Maybe someone has a nice tutorial to learn how I can use this model to predict my target variable. I have to use this class and can't use Pyro or any other framework currently, because otherwise I can't use the othe…