[enhance] Replace column selectors with sklearn funcs
nabenabe0928 opened this issue · 0 comments
nabenabe0928 commented
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)