fix

Samsung · Jan 17, 2025 · fdc8491 · fdc8491
1 parent d3b3c87
commit fdc8491
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/experiment/src/lstm_model.py b/experiment/src/lstm_model.py
@@ -77,11 +77,10 @@ def build(self, hp=None) -> Model:
         dropout_dense_b = Dropout(dropout_b, name="b_dropout")(dense_b)
 
         dense_final = Dense(units=1, activation='sigmoid', name="prediction", dtype=self.d_type)(dropout_dense_b)
-        output = Dropout(dropout_final, name="final_dropout")(dense_final)
 
         metrics = [BinaryAccuracy(name="binary_accuracy"), Precision(name="precision"), Recall(name="recall")]
 
-        model: Model = Model(inputs=[line_input, variable_input, value_input, feature_input], outputs=output)
+        model: Model = Model(inputs=[line_input, variable_input, value_input, feature_input], outputs=dense_final)
         model.compile(optimizer=Adam(), loss='binary_crossentropy', metrics=metrics)
         model.summary(line_length=120, expand_nested=True, show_trainable=True)