TeamHG-Memex / eli5

A library for debugging/inspecting machine learning classifiers and explaining their predictions
http://eli5.readthedocs.io
MIT License
2.74k stars 332 forks source link

assert fails when trying to explain lightgbm model #339

Open ophiry opened 4 years ago

ophiry commented 4 years ago

https://github.com/TeamHG-Memex/eli5/blob/4cd66ac5e0ffde0c002d8f46dc64d7091f49445f/eli5/lightgbm.py#L252

pred_leafs contains num_trees * len(X) values tree_info contains num_trees values

so the values are identical only when there's a single data row to explain