automl / automl/Auto-PyTorch

[enhance] Replace column selectors with sklearn funcs

Open
#393 0 comments 0 reactions 0 assignees View on GitHub
enhancement not urgent
Dominant language
Python
Stars
2.5k
Forks
300
PR merge metrics
No merged PRs in 30d

Description

Currently, we are picking columns by ourselves, but since sklearn has [its own column selectors](https://inria.github.io/scikit-learn-mooc/python_scripts/03_categorical_pipeline_column_transformer.html), we can use it.

The selectors for numerical and categorical columns can be simply implemented by the following:
```python
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)
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.