lightgbm-ruby
lightgbm-ruby copied to clipboard
ankane
High performance gradient boosting for Ruby
Next - [ ] Support different prediction types - [ ] Add `feature_importance_type` option 0.2.0 - [x] Add support for LightGBM 3.0 (just waiting for final release) - `edge` branch...
Given the popular usage of hashes in Ruby, I think this would be a nice quality of life addition to the Booster#predict API.
Adds support to have categorical features label encoded automatically, when using Booster#predict. The python api does it by relying on pandas dataframes: https://github.com/microsoft/LightGBM/blob/e057ae08e6bf6c6c84f276a127423fb145ca5fdb/python-package/lightgbm/basic.py#L1131-L1137 Would be great for this ruby lib...