더북(TheBook)

4.1.5 훈련된 모델로 새로운 데이터에 대해 예측하기

모델을 훈련시킨 후 이를 실전 환경에서 사용하고 싶을 것입니다. 3장에서 배웠듯이 predict 메서드를 사용해서 어떤 리뷰가 긍정일 확률을 예측할 수 있습니다.14

>>> model.predict(x_test)
array([[ 0.98006207]
       [ 0.99758697]
       [ 0.99975556]
       ...,
       [ 0.82167041]
       [ 0.02885115]
       [ 0.65371346]], dtype=float32)

여기처럼 이 모델은 어떤 샘플에 대해 확신을 가지고 있지만(0.99 또는 그 이상, 0.01 또는 그 이하) 어떤 샘플에 대해서는 확신이 부족합니다(0.6, 0.4).

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