search

sberbank-ai-lab / LightAutoML

LAMA - automatic model creation framework
Apache License 2.0
887 stars 92 forks source link

ReportDeco parameter typo #125

Closed se-teryoshkin closed 2 years ago

se-teryoshkin commented 2 years ago

https://github.com/sberbank-ai-lab/LightAutoML/blob/master/lightautoml/report/report_deco.py#L458

Default parameter is ton_n_classes while actually L931 and L933 are referencing to top_n_classes. They are raising KeyError exception.

VaBun commented 2 years ago

Fixed. Thank you very much for noticing!