본문 바로가기
2/[ Machine Learning ]

Keras Initializer

by Kieran_Han 2020. 12. 23.

keras.initializers 종류

 

keras는 기본적으로 균등분포의 Glorot 초기화를 사용한다.

다음과 같이 층을 만들 때, 

kernel_initializer = "he_uniform" 이나 kernel_initializer = "he_normal"로 바꾸어 He 초기화를 사용할 수 있다.

 

fan_in 대신 fan_out 기반의 균등분포 He 초기화를 사용하고 싶다면 다음과 같이 Variance Scaling을 사용할 수 있다.