[Deep Learning] 분류 문제 Loss 셋팅

📝딥러닝 분류 문제 Loss 셋팅

2개로 분류하는 문제에서의 로스펑션은 bibary_crossentropy를 사용하고
3개 이상으로 분류하는 문제에는 두가지 방법이 있는데 y의 값을 확인하여 선택한다

첫번째 방법은 y값이 레이블 인코딩으로 되어있으면 sparse_categorical_crossentropy
두번째 방법은 y값이 원핫인코딩으로 되어있으면 categorical_crossentropy

def build_model() :
  model = Sequential()
  model.add( Flatten() )
  model.add( Dense( units= 128, activation= 'relu') )
  model.add( Dense( units= 64, activation= 'relu') )
  model.add( Dense( units=10, activation= 'softmax' ) )

  # y값이 레이블 인코딩으로 되어있으므로 sparse_categorical_crossentropy 사용
  model.compile( optimizer= 'adam', loss= 'sparse_categorical_crossentropy', metrics= [ 'accuracy' ])
  return model