sberbank-ai-lab / LightAutoML

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

LabelEncoder filtering is not working #105

Closed se-teryoshkin closed 2 years ago

se-teryoshkin commented 2 years ago

https://github.com/sberbank-ai-lab/LightAutoML/blob/master/lightautoml/transformers/categorical.py#L167

I suppose this line contains an error, because this line tries to filter an entire dataframe using integer value.

I have copy-pasted this code fragment and added few prints: image

As we are able to see, there is no filtering.

If we fix it by adding a column to the dataframe, it will work well. image