From fdc8491346045745ece213feb91266b7b6d57393 Mon Sep 17 00:00:00 2001 From: Roman Babenko Date: Fri, 17 Jan 2025 22:35:58 +0200 Subject: [PATCH] fix --- experiment/src/lstm_model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/experiment/src/lstm_model.py b/experiment/src/lstm_model.py index 2fdcefdb9..1eaf01991 100644 --- 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)