5 평가하기

예측 결과는 각 클래스에 대한 확률값을 반환한다. softmax는 이 확률값을 더했을 때 1이 되는 값을 출력하는데 여기에서 가장 큰 값을 예측값으로 사용한다. 이를 위해 np.argmax로 가장 큰 값의 인덱스를 반환한 값(클래스 예측)을 y_predict에 할당한다.

y_predict = np.argmax(y_pred, axis=1)
y_predict[:10]

실행 결과

array([2, 2, 2, 0, 0, 2, 2, 0, 2, 2])

np.argmax를 이용해 가장 큰 값의 인덱스를 반환한 값(클래스 예측)을 y_test_val에 할당한다.

y_test_val = np.argmax(y_test.values, axis=1)