[넘파이] axis 옵션 활용한 argmax 함수 사용, 최대값 위치

[넘파이] argmax 최대값 위치 추출 axis 옵션 활용¶

• 어레이에서 가장 큰 값의 위치를 알려줍니다.

In [2]:

import numpy as np
data = np.array([0, 2, 3, 10, 3, 2, 1])
np.argmax(data)

Out[2]:

3

• axis 옵션을 활용해서 지정된 차원 별 가장 큰 값의 위치를 알려줍니다.

In [3]:

data = np.array([[1, 2, 3],  # 1, 2, 3 중 가장 큰값의 위치 : 2
                 [6, 5, 4],  # 6, 5, 4 중 가장 큰값의 위치 : 0
                 [8, 9, 7]]) # 8, 9, 7 중 가장 큰값의 위치 : 1

np.argmax(data, axis=1)

Out[3]:

array([2, 0, 1])

• 분류 문제를 해결하면 predict proba 값이 도출되고 이중 최종 클래스를 결정하기 위해 np.argmax를 많이 사용하게 되네요 !
• 내 분류모델의 배치 사이즈가 3이라면 아래와 같은 axis옵션을 설정해 주어야 합니다 !
• 파이토치이 torch.argmax를 사용하면 되긴하지만 tensor를 array로 전환하여 사용하기도 합니다.