더북(TheBook)

이제 모델의 하이퍼파라미터 값을 정의하겠습니다.

코드 10-54 하이퍼파라미터 초기화

VOCAB_LENGTH = len(tokenizer.vocab)
EMB_DIM = 200
CNN_FILTERS = 100
DNN_UNITS = 256
OUTPUT_CLASSES = 2
DROPOUT_RATE = 0.2
NB_EPOCHS = 5

TEXT_MODEL 네트워크에 마지막 단계에서 정의한 하이퍼파라미터 값을 전달합니다.

코드 10-55 하이퍼파라미터 값을 네트워크에 전달

text_model = TEXT_MODEL(vocabulary_size=VOCAB_LENGTH,
                        embedding_dimensions=EMB_DIM,
                        cnn_filters=CNN_FILTERS,
                        dnn_units=DNN_UNITS,
                        model_output_classes=OUTPUT_CLASSES,
                        dropout_rate=DROPOUT_RATE)
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.