Auto-PyTorch icon indicating copy to clipboard operation
Auto-PyTorch copied to clipboard

[enhance] Replace column selectors with sklearn funcs

Open nabenabe0928 opened this issue 3 years ago • 0 comments

Currently, we are picking columns by ourselves, but since sklearn has its own column selectors, we can use it.

The selectors for numerical and categorical columns can be simply implemented by the following:

import numpy as np
from sklearn.compose import make_column_selector as selector

numerical_col_selector = selector(dtype_include=np.number)
categorical_col_selector = selector(dtype_exclude=np.number)

nabenabe0928 avatar Feb 28 '22 13:02 nabenabe0928