search

cog-imperial / entmoot

Multiobjective black-box optimization using gradient-boosted trees
https://entmoot.readthedocs.io/
BSD 3-Clause "New" or "Revised" License
56 stars 12 forks source link

ENTMOOT does not work with newest lightgbm version #29

Closed spiralulam closed 5 months ago

spiralulam commented 12 months ago

To reproduce it, try to run notebook "multi_obj_with_constraints" with the newest version of lightgbm.

TobyBoyne commented 5 months ago

After looking into it, I found the root of the issue - there was an undocumented change from v4.0.0 -> v4.1.0, where the categorical_feature argument of Dataset only accepts lists, and no longer allows tuples. Converting cat_idx to a list when fitting the model fixes this issue.