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을 대입해주면 된다.
'인공지능 > 텐서플로우' 카테고리의 다른 글
CNN의 convolution과 pooling (0) | 2022.06.14 |
---|---|
분류의 문제에서 loss 함수를 categorical_crossentropy로 설정할 상황 (0) | 2022.06.14 |
텐서플로우의 모델 중, 네트워크와 웨이트를 따로 저장하고 불러오는 방법 (0) | 2022.06.14 |
텐서플로우의 모델을 저장하고 불러오는 방법 (0) | 2022.06.14 |
validation_data 파라미터 사용법 (0) | 2022.06.14 |