sberbank-ai-lab/LightAutoML

LabelEncoder filtering is not working

se-teryoshkin opened this issue · 0 comments

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