본문 바로가기

Deep Learning/Tensorflow

Tensorflow - activation function [ softmax ]

반응형
softmax

 

여러 클래스로 분류할때,

 

아웃풋 레이어 노드의 숫자는 분류해야하는 숫자에 맞춰서 놓게되는데


softmax는 3개 이상의 분류문제 activation function으로 사용한다.

 

 

사용코드 예시

 

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'))

 

반응형