from tensorflow.keras.utils import to_categorical
train_y = to_categorical(train_y)
test_y = to_categorical(test_y)
print(train_y.shape, test_y.shape)
'python > tensorflow,keras' 카테고리의 다른 글
| tensorflow.keras.callbacks (0) | 2022.10.01 |
|---|---|
| 학습한 모델 저장 (0) | 2022.10.01 |
| keras ImageDataGenerator (0) | 2022.10.01 |
