28b8b9d39f53d8327dbf658048a81b7046ae398f,magpie/nn/models.py,,rnn,#Any#Any#,56

Before Change

    model.add(BatchNormalization())
    model.add(Dropout(0.1))

    model.add(Dense(output_length, activation="sigmoid"))

    model.compile(
        loss="binary_crossentropy",

After Change

     Create and return a keras model of a RNN 
    HIDDEN_LAYER_SIZE = 256

    inputs = Input(shape=(SAMPLE_LENGTH, embedding_size))

    gru = GRU(
        HIDDEN_LAYER_SIZE,
        input_shape=(SAMPLE_LENGTH, embedding_size),
        kernel_initializer="glorot_uniform",
        recurrent_initializer="normal",
        activation="relu",
    )(inputs)

    batch_normalization = BatchNormalization()(gru)
    dropout = Dropout(0.1)(batch_normalization)
    outputs = Dense(output_length, activation="sigmoid")(dropout)

    model = Model(inputs=inputs, outputs=outputs)

    model.compile(
        loss="binary_crossentropy",
        optimizer="adam",

Instances