📝딥러닝 드랍아웃
드랍아웃이란 뉴런에 연결된 선을 일부분을 잘라서 학습이 잘 되도록 하는 방법이다
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
'인공지능 > Deep Learning' 카테고리의 다른 글
[Deep Learning] TensorFlow 레이블 인코딩된 값을 원핫 인코딩으로 바꾸기 (0) | 2022.12.29 |
---|---|
[Deep Learning] TensorFlow 모델 저장 / 불러오기 (0) | 2022.12.29 |
[Deep Learning] DNN TensorFlow 이미지 분류하는 딥러닝 해보기 (0) | 2022.12.29 |
[Deep Learning] epochs 횟수 조정 (0) | 2022.12.29 |
[Deep Learning] 분류 문제 Loss 셋팅 (0) | 2022.12.28 |
댓글