AxeldeRomblay / MLBox

MLBox is a powerful Automated Machine Learning python library.
https://mlbox.readthedocs.io/en/latest/
Other
1.49k stars 274 forks source link

How to specify categorical columns #102

Closed MasterXiong closed 4 years ago

MasterXiong commented 4 years ago

Hi, thanks for providing such a useful toolkit. I want to ask that when we use Optimizer.optimise(space, data), how should we setup the input data so that mlbox could know which columns are categorical features that need to apply the encoding strategy? Thanks.

AxeldeRomblay commented 4 years ago

Hello @MasterXiong ! You don't need to specify which columns are categorical features (neither dates nor lists nor whatever...), MLBox will detect automatically :)