bonsai icon indicating copy to clipboard operation
bonsai copied to clipboard

performance bug with `engine = "lightgbm"`

Open simonpcouch opened this issue 11 months ago • 2 comments

Noticed while working on emlwr the other day that bonsai::train_lightgbm() is quite a bit slower than lightgbm::lgb.train(), probably due to the handling of categorical variables / conversion to lgb.Dataset. Observed with emlwr:::simulate_classification()—@EmilHvitfeldt also noted a slowdown from a user with a similar-appearing dataset last week.

simonpcouch avatar Nov 12 '24 15:11 simonpcouch