From 96092fd91034c62ea8daa384b74fe1c8dfaf5a13 Mon Sep 17 00:00:00 2001 From: Tyler Morrow Date: Wed, 2 Oct 2024 12:17:05 -0600 Subject: [PATCH] Fix use of dropout in . --- riid/models/neural_nets/basic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/riid/models/neural_nets/basic.py b/riid/models/neural_nets/basic.py index 48053d6..705b32e 100644 --- a/riid/models/neural_nets/basic.py +++ b/riid/models/neural_nets/basic.py @@ -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)