더북(TheBook)

4 예측하기

 

predict() 메서드로 모델을 예측해 보자. predict() 메서드는 시험 데이터의 텍스트가 어떤 분류에 속하는지 예측한다. 이제 모델이 정답을 예측하고 성적표를 받을 시기가 된 것이다.

y_pred = model.predict(X_test_sp)
y_pred[:10]

실행 결과

14/14 [==============================] - 5s 223ms/step
array([[0.04029479, 0.01259641, 0.94710886],
       [0.19934632, 0.33347803, 0.46717563],
       [0.27292505, 0.01636682, 0.7107081 ],
       [0.9922458 , 0.0032146 , 0.00453948],
       [0.9856799 , 0.00695921, 0.00736091],
       [0.19694512, 0.04934863, 0.7537062 ],
       [0.17981924, 0.05678767, 0.76339316],
       [0.9838552 , 0.00562436, 0.01052044],
       [0.05183903, 0.01426377, 0.9338971 ],
       [0.04421081, 0.10807092, 0.84771824]], dtype=float32)
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.