search

automl / HPOlib1.5

GNU General Public License v3.0
69 stars 27 forks source link

Unable to load dependencies for BNN examples #77

Closed Neeratyoy closed 5 years ago

Neeratyoy commented 5 years ago

Hi,

I am trying to work on this example where BostonHousing is loaded using, from hpolib.benchmarks.ml.bnn_benchmark import BNNOnBostonHousing

However, this imports BayesianNeuralNetwork from the sgmcmc package. The sgmcmc package seems to have just one branch with no class definition for BayesianNeuralNetwork.

As a result, from hpolib.benchmarks.ml.bnn_benchmark import BNNOnBostonHousing gives the error, ImportError: cannot import name 'BayesianNeuralNetwork'

Am I missing something here? How can I solve this?

Thanks.

KEggensperger commented 5 years ago

Hi, Yes, the package you referenced does not have the BNN interface you need. You need to use a different fork from here pip install git+https://github.com/automl/sgmcmc.git

Neeratyoy commented 5 years ago

Thanks a lot! That worked!