search

automl / RoBO

RoBO: a Robust Bayesian Optimization framework
BSD 3-Clause "New" or "Revised" License
480 stars 134 forks source link

What if the parameters must be integer? #105

Closed jiahuanluo closed 5 years ago

jiahuanluo commented 5 years ago

Can I set robo.fmin.bayesian_optimization to select params in integer. i.e. the param is in a range of [1, 2, 3, 4, 5]

aaronkl commented 5 years ago

RoBO cannot handle integer parameters by itself. Instead, you can defining the bounds as [1,5] and then inside your objective function round it to an integer. Obviously this is somewhat suboptimal since different points in your input space lead to the same integer parameter, but RoBO's model should eventually figure that out.

jiahuanluo commented 5 years ago

Thank you! I will try to compare it with SMAC.