sklearn | PolynominalFeatures 사용하기

PolynominalFeatures는 입력으로 들어온 2차원 배열 x를 다항식으로 변환하며, 다음과 같이 사용할 수 있다.

from sklearn.preprocessing import PolynomialFeatures
poly = PolynomialFeatures(degree=2, interaction_only=False, include_bias=False)

지정된 차수(degree)보다 작거나 같은 feature들의 다항식 조합을 만든다.

• degree: 차수(기본값 2)
• interaction_only: 상호작용항 포함 여부
  • True이면, 제곱 항 제외 상호작용항까지만
  • False이면, 모든 항(기본값)
• include_bias: 상수항(0제곱) 포함 여부
  • True이면, 상수항 포함(기본값)
  • False이면, 상수함 불포함

 

위 코드를 직접 실행해보면 다음과 같은 결과가 나온다.

poly의 입력으로 x=[[1], [2]] 주어지면, 결과는 [[$1, 1\times2], [1^2, 2^2$]]으로 2차항까지 모든 항이 출력된다.

 

interaction_only=True이면 결과는 다음과 같다.

제곱항은 제외하고 기본 상수항 1과 상호작용항 $1\times2$만 결과로 나타난다.

 

이 상태에서 include_bias=True가 되면 어떻게 되는지 살펴보면 다음과 같다.

0제곱 항이 추가되어 $[[1^0, 1], [2^0, 1\times2]]$이 결과로 출력된 것을 볼 수 있다. 

 

 

sklearn.preprocessing.PolynomialFeatures