Skip to content

Commit

Permalink
Fix use of dropout in .
Browse files Browse the repository at this point in the history
  • Loading branch information
tymorrow committed Oct 2, 2024
1 parent efa4d72 commit 96092fd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion riid/models/neural_nets/basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ def fit(self, ss: SampleSet, batch_size: int = 200, epochs: int = 20,
kernel_regularizer=l2(self.l2_alpha),
)(inputs)
if self.dropout is not None:
last_layer = Dropout(0.2)(dense_layer)
last_layer = Dropout(self.dropout)(dense_layer)
else:
last_layer = dense_layer
outputs = Dense(Y.shape[1], activation=self.final_activation)(last_layer)
Expand Down

0 comments on commit 96092fd

Please sign in to comment.