군침이 싹 도는 코딩

레이블 인코딩 된 y 값을 원 핫 인코딩으로 바꾸기 본문

Python/Deep Learning

레이블 인코딩 된 y 값을 원 핫 인코딩으로 바꾸기

mugoori 2022. 12. 29. 17:02
y_train
>>> array([5, 0, 4, ..., 5, 6, 8], dtype=uint8)

y_train = tf.keras.utils.to_categorical(y_train, num_classes= 10 )

y_train
>>>
array([[0., 0., 0., ..., 0., 0., 0.],
       [1., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       ...,
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 0., 0.],
       [0., 0., 0., ..., 0., 1., 0.]], dtype=float32)

# 레이블 인코딩 된 y값을 손쉽게 원핫 인코딩으로 바꿀 수 있는 방법이 있다

텐서플로우/케라스/유틸/투 카테고리컬을 사용해 바꿀 변수와 그 변수의 아웃풋 노드를 적어주면 된다