Commit 6688a01f authored by lucas_miranda's avatar lucas_miranda
Browse files

Fixed bug in hyperparameter tuning

parent 3e4a5e3d
Pipeline #86897 failed with stage
in 15 minutes and 48 seconds
......@@ -526,9 +526,9 @@ class SEQ_2_SEQ_GMVAE:
encoder = Model_E4(encoder)
encoder = BatchNormalization()(encoder)
encoding_shuffle = deepof.model_utils.MCDropout(self.DROPOUT_RATE)(encoder)
# encoding_shuffle = deepof.model_utils.MCDropout(self.DROPOUT_RATE)(encoder)
z_cat = Dense(self.number_of_components, activation="softmax",)(
encoding_shuffle
encoder
)
z_cat = deepof.model_utils.Entropy_regulariser(self.entropy_reg_weight)(z_cat)
z_gauss = Dense(
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment