더북(TheBook)

마침내 LogisticRegression 추정기가 StandardScalerPCA로 변환된 훈련 데이터로 학습합니다. 다시 언급하지만 파이프라인의 중간 단계 횟수는 제한이 없습니다. 하지만 파이프라인을 예측 작업에 사용하려면 파이프라인의 마지막 요소는 추정기가 되어야 합니다.

파이프라인의 마지막 단계가 추정기라면 파이프라인에서 fit 메서드를 호출하는 것과 비슷하게 predict 메서드도 제공합니다. Pipeline 인스턴스의 predict 메서드를 호출할 때 주입된 데이터는 중간 단계의 transform 메서드를 통과합니다. 마지막 단계에서 추정기 객체가 변환된 데이터에 대한 예측을 반환합니다.

사이킷런의 파이프라인은 매우 유용한 래퍼 도구입니다. 책 나머지 부분에서 자주 사용하게 될 것입니다. Pipeline 객체가 동작하는 방식을 확실하게 이해하기 위해 그림 6-1을 자세히 살펴보세요. 이전 문단에서 설명한 것을 요약했습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.