Data science/머신러닝

feature engineering(특성 공학)

gokimkh 2022. 7. 15. 20:23

개념

기존의 특성을 사용해 새로운 특성을 뽑아내는 작업

 

문법

사이킷런의 변환기를 사용한다.

from sklearn.preprocessing import PolynomialFeatures    # 사이킷런 변환기 사용

poly = PolynomialFeatures()
poly.fit([[2,3]])   # 2개의 특성: 2 3
print(poly.transform([[2,3]]))

poly = PolynomialFeatures(include_bias=False)    # 1 특성 삭제하기
poly.fit([[2,3]])   # 2개의 특성: 2 3
print(poly.transform([[2,3]]))

# x0은 첫 번째 특성, x1는 두번째 특성을 의미
poly.get_feature_names_out()