더북(TheBook)

Note ≡


역주 훈련 데이터셋을 전처리할 때 사용한 파라미터로 테스트 데이터셋을 변환해야 합니다. 비슷하게 다음 절에서 소개할 교차 검증도 훈련 폴드에서 학습한 파라미터를 사용하여 테스트 폴드를 전처리해야 합니다. 이렇게 하려면 전처리 과정이 교차 검증 반복 안으로 들어가야 합니다. 전처리 과정을 Pipeline 클래스로 감싸서 교차 검증 함수로 전달하면 손쉽게 구현할 수 있습니다.

사이킷런 0.23 버전에서 추정기 객체를 주피터 노트북에서 시각화해 주는 기능이 추가되었습니다. 특히 파이프라인 구조를 시각적으로 잘 요약해 주며 다음과 같이 사용합니다.

>>> from sklearn import set_config
>>> set_config(display='diagram')
>>> pipe

▲ 그림 6-2 주피터 노트북에 시각화된 사이킷런 추정기

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