search

sb-ai-lab / LightAutoML

Fast and customizable framework for automatic ML model creation (AutoML)
https://developers.sber.ru/portal/products/lightautoml
Apache License 2.0
1.12k stars 52 forks source link

lime class mapping bug #8

Closed DESimakov closed 2 years ago

DESimakov commented 2 years ago

https://github.com/sb-ai-lab/LightAutoML/blob/9745796d9fe7f74ed740889ee1b882c49ee5304c/lightautoml/addons/interpretation/lime.py#L304 Replace on: if (class_names is None) or (len(class_names) == 0):