AI·빅데이터 융합 경영학 Study Note

scikit-learn의 make_column_selector로 범주형 수치형 자동 구분 본문

AI·ML

scikit-learn의 make_column_selector로 범주형 수치형 자동 구분

SubjectOwner 2024. 10. 24. 20:46

scikit-learn의 make_column_selector 기능을 사용하여 특정 데이터 유형을 기준으로 열을 선택하는 방법을 보여줍니다. 이를 통해 데이터 전처리에서 범주형 또는 수치형 데이터를 쉽게 구분하고 처리할 수 있습니다. 

from sklearn.compose import make_column_selector

cat_selector = make_column_selector(dtype_include=object)
num_selector = make_column_selector(dtype_include=np.number)


#num_selector(X)
#cat_selector(X)