본문 바로가기
인공지능/Deep Learning

[Deep Learning] Dropout

by coding_su 2022. 12. 29.

📝딥러닝 드랍아웃

드랍아웃이란 뉴런에 연결된 선을 일부분을 잘라서 학습이 잘 되도록 하는 방법이다

 

from keras.layers import Dropout

def build_model() :
  model = Sequential()
  model.add( Dense( units= 128, activation= 'relu', input_shape=(784, ) ) )
  model.add( Dropout(0.2) ) # rate 값 지정 랜덤으로 이 구간의 선을 20% 없애라는 뜻
  model.add( Dense( units= 64, activation= 'relu') )
  model.add( Dense( units=10, activation= 'softmax' ) )
  model.compile( optimizer= 'adam', loss= 'sparse_categorical_crossentropy', metrics= [ 'accuracy' ])
  return model

 

댓글