model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(128, 'relu'))
model.add(tf.keras.layers.Dense(10, 'softmax'))
model.compile('adam', loss = 'sparse_categorical_crossentropy', metrics = ['accuracy'])
model.fit(training_images, training_labels, epochs = 5)
Softmax(소프트맥스)는 입력받은 값을 출력으로 0~1사이의 값으로 모두 정규화하며 출력 값들의 총합은 항상 1이 되는 특성을 가진 함수이다. 출력값은 각 레이블의 확률이 된다.
'인공지능 > 텐서플로우' 카테고리의 다른 글
softmax로 나온 결과를 레이블 인코딩으로 바꾸는 방법 (0) | 2022.06.13 |
---|---|
분류의 문제에서 loss 셋팅하는 방법 (0) | 2022.06.13 |
Flatten 라이브러리 사용하는 이유 (0) | 2022.06.13 |
EarlyStopping 라이브러리 사용법 (0) | 2022.06.13 |
validation 데이터란 무엇이고 코드에서 사용하는 방법 (validation_split) (0) | 2022.06.13 |