2017-03-25 2 views
2

파이썬 NumPy와 하나의 핫 지역

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

등이 하나의 뜨거운 인코딩 매트릭스

array([[[1, 0, 0], 
     [1, 0, 0], 
     [0, 1, 0]], 

     [[0, 0, 1], 
     [0, 1, 0], 
     [1, 0, 0]]]) 

을 할 수있는 가장 좋은 방법은 무엇에, 어떻게 핫 배열을 디코딩?

+0

당신이 뒤에 논리를 설명해 주시겠습니까? '디코드'란 무엇을 의미합니까? – kmario23

+0

@ kmario23 하나의 핫 인코딩의 역 동작. 이 질문은 이미 아래에 답변되어 있습니다. –

+0

One Hot로 인코딩 된 행렬과 아래의 두 번째 행렬을 작성하여 One Hot에서 역전되었습니다. 이 작업을 수행하는 데 필요한 코드가 생겼습니다. –

답변

1

사용 np.argmaxaxis=2 함께 -

a.argmax(2) 

샘플 실행 -

In [186]: a 
Out[186]: 
array([[[1, 0, 0], 
     [1, 0, 0], 
     [0, 1, 0]], 

     [[0, 0, 1], 
     [0, 1, 0], 
     [1, 0, 0]]]) 

In [187]: a.argmax(2) 
Out[187]: 
array([[0, 0, 1], 
     [2, 1, 0]]) 
관련 문제