Note ≡

역주 사이킷런에서 제공하는 순차 특성 선택 알고리즘인 SequentialFeatureSelector 클래스를 Wine 데이터셋에 적용해 보겠습니다.

이 코드는 https://github.com/rickiepark/ml-with-pytorch/blob/main/ch04/SequentialFeature Selector.ipynb에 있습니다.

먼저 SequentialFeatureSelector를 임포트합니다.

>>> from sklearn.feature_selection import SequentialFeatureSelector

사용할 모델 객체를 첫 번째 매개변수로 전달합니다. 선택할 특성의 개수는 n_features_to_select에서 지정합니다. 기본값은 입력 특성의 절반입니다. 0~1 사이 실수를 지정하면 선택할 특성의 비율로 인식합니다.

direction 매개변수로 특성 선택 방향(전진 또는 후진)을 선택합니다. 기본값은 전진을 의미하는 'forward'고 후진을 선택하려면 'backward'로 지정합니다.

SequentialFeatureSelector는 특성을 선택하기 위해 교차 검증을 사용합니다. cv 매개변수에서 교차 검증 횟수를 지정할 수 있습니다. 기본값은 5입니다. 회귀 모델일 경우 KFold, 분류 모델일 경우 StratifiedKFold를 사용하여 폴드를 나눕니다. 두 클래스에 대한 자세한 내용은 6장을 참고하세요.