search

linkedin / FastTreeSHAP

Fast SHAP value computation for interpreting tree-based models
BSD 2-Clause "Simplified" License
512 stars 31 forks source link

[BUG] LightGBM Booster 'objective' Parameter Membership with `shortcut=True` #4

Closed ngupta20 closed 2 years ago

ngupta20 commented 2 years ago

Note This corresponds to an issue on slundberg/shap.

Description A LightGBM Booster does not necessarily have 'objective' in its param attribute dictionary. The documented default behavior in LightGBM is to assume regression without auto-setting the 'objective' key in the Booster.params dictionary if 'objective' is missing from param (a dictionary passed into the Training API function train).

Reproduction The following is a link to a notebook containing details for error reproduction and a view of the stack trace: Notebook for Reproduction

Thanks for your responsiveness!

Nandish Gupta Data Science Engineer, SolasAI image