소프트맥스(Softmax)란?

1. 소프트맥스 함수(Softmax function) 란?

입력값을 0에서 1사이의 값으로 정규화하여 k개의 클래스에 대한 확률을 추정하는 함수

* 이때 출력값들의 총합은 1이 된다.

 

2. 사용법

Label	Description
0	T-shirt/top
1	Trouser
2	Pullover
3	Dress
4	Coat
5	Sandal
6	Shirt
7	Sneaker
8	Bag
9	Ankle boot

 

위와 같이 분류하고 싶은 클래수의 수만큼 10개의 출력으로 구성한다.

 

# output layer
model.add(tf.keras.layers.Dense(10, 'softmax'))

 

학습 후 예측 데이터

 

 

# 10개의 값 중에서 가장 큰 값을 선택하게 하여 0 ~ 9까지 예측한 값을 출력하도록 한다.
y_pred = y_pred.argmax(axis=1)