np.array 내 최대 값을 갖는 인덱스
import numpy as np np.argmax(pred)