개발공부

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

Python/Deep Learning

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

mscha 2022. 6. 14. 14:13

y는 0 - 9의 값으로 레이블 인코딩되어 있다. 

>>> y_train
array([5, 0, 4, ..., 5, 6, 8], dtype=uint8)

이를 원핫 인코딩된 값으로 바꾸는 방법은 아래와 같다.

to_categorical(인코딩할 값, num_classes = 분류 개수)

from tensorflow.keras.utils import to_categorical
y_train = to_categorical(y_train, num_classes = 10)