인공지능/텐서플로우

레이블링된 y값을 to_categorical 함수 이용해서 원핫인코딩으로 바꾸기

iminu 2022. 6. 14. 14:04
from tensorflow.keras.utils import to_categorical
y_test = to_categorical(y_test, 10)

레이블링된 y_test는 10가지 값을 가지고 있다. 

to_categorical(y_test, 10) 원핫인코딩으로 바꿀 데이터 y_test와 범주 10을 대입해주면 된다.