더북(TheBook)

5 판별자 신경망 만들기

앞에서는 생성자 신경망을 만들었습니다. 생성자 신경망은 어떤 숫자 이미지를 만들어 내는 신경망입니다. 하지만 판별자 신경망은 이미지를 만들어 내는 신경망이 아니라 생성자 신경망이 만든 이미지가 가짜인지를 정확하게 판별하는 신경망입니다.

이 신경망 역시 훈련되지 않은 신경망입니다. 앞으로 적대적 생성 신경망(GAN)을 통해 생성자 신경망과 지금 만드는 판별자 신경망을 학습시켜 나갈 예정입니다.

def create_discriminator():
    discriminator = Sequential()
    discriminator.add(Dense(units=512,input_dim=784))
    discriminator.add(LeakyReLU(0.2))
    discriminator.add(Dense(units=256))
    discriminator.add(LeakyReLU(0.2))
    discriminator.add(Dense(units=1, activation='sigmoid'))
    discriminator.compile(loss='binary_crossentropy',
optimizer = Adam(lr=0.0002, beta_1=0.5))
    return discriminator
d = create_discriminator()
d.summary()
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.