search

JavierAntoran / Bayesian-Neural-Networks

Pytorch implementations of Bayes By Backprop, MC Dropout, SGLD, the Local Reparametrization Trick, KF-Laplace, SG-HMC and more
MIT License
1.83k stars 302 forks source link

Energy Efficient dataset #11

Closed caxelrud closed 4 years ago

caxelrud commented 4 years ago

For bbp_hetero notebook, the Energy Efficiency dataset has 8 inputs and 2 outputs. At the moment, output 1 is being used as an input to model output 2. You can just drop the last column of the dataset and model 8 inputs and 1 output. I tried and worked fine.

JavierAntoran commented 4 years ago

Sure. This choice was made because generally, in the BNN literature, the one-output version of the dataset is used.