더북(TheBook)

마지막으로 샘플 하나의 클래스 레이블을 예측할 때 주의할 점이 있습니다. 사이킷런은 입력 데이터로 2차원 배열을 기대합니다. 하나의 행을 2차원 포맷으로 먼저 변경해야 합니다. 하나의 행을 2차원 배열로 변환하는 한 가지 방법은 넘파이 reshape 메서드를 사용하여 새로운 차원을 추가하는 것입니다.13 예를 들어 다음과 같습니다.

>>> lr.predict(X_test_std[0, :].reshape(1, -1))
array([2])

 

 


13 역주 다른 방법으로는 np.expand_dims(X_test_std[0, :], axis=0)처럼 사용할 수 있습니다. 가장 간단한 방법은 X_test_std [0:1, :]처럼 인덱스 슬라이싱을 사용하는 것입니다.

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