• 카테고리

    질문 & 답변
  • 세부 분야

    딥러닝 · 머신러닝

  • 해결 여부

    미해결

Dense 레이어의 units 질문입니다!

22.12.21 06:57 작성 조회수 918

0

안녕하세요.

좋은 강의 열심히 듣고 있습니다.

 

선생님께서 시퀀셜 모델의 히든 레이어에

 

Dense(32, input_shape ...)

Dense(16, ...)

Dense(1, ...)

 

이런 식으로 3개의 레이어를 추가하셨는데,

32나 16이라는 값은 어디서 왔는지 잘 모르겠습니다.

 

임의대로 해도 되는 건지, 값을 크게 줄 때와 작게 줄 때의 장단점 등.. 궁금합니다

 

감사합니다!

 

답변 1

답변을 작성해보세요.

0

어떤 규칙이 있는 것은 아니고 경험적인 임의의 값입니다. 32와 16은 은닉층의 neuron 갯수인데 이 갯수가 많으면 neuron 간의 연결 갯수가 많아지는데 이 것은 학습할 weight의 갯수가 많아진다는 뜻이므로 많은 더 학습 data가 필요합니다. 요약하면 학습할 데이터가 많으면 neuron의 갯수를 더 많이 지정 (예, 64, 128 등)하고 은닉층 (hidden layer)수를 더 늘려도 되고, 학습할 데이터가 적으면 weight수를 줄이는 것이 좋으므로 neuron 수와 은닉층 수를 줄이는 것이 좋습니다. 좋은 질문 감사합니다.