Skip to content

[enhance] Replace column selectors with sklearn funcs #393

Open
@nabenabe0928

Description

@nabenabe0928

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)

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requestnot urgentThis pull request does not need immediate attention.

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions