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)
    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.