From f3ce92d931636ca6b94d72279d6e4ee548a361d8 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Sun, 12 Aug 2018 11:30:43 +0200
Subject: [PATCH] allow both None and 0 for deactivating dropout
---
toolkit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index a318c21..3ef7399 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -623,7 +623,7 @@ class ClassificationProject(object):
# the (other) hidden layers
for node_count, dropout_fraction in zip(self.nodes[start_layer:], self.dropout[start_layer:]):
self._model.add(Dense(node_count, activation=self.activation_function))
- if dropout_fraction > 0:
+ if (dropout_fraction is not None) and (dropout_fraction > 0):
self._model.add(Dropout(rate=dropout_fraction))
# last layer is one neuron (binary classification)
self._model.add(Dense(1, activation=self.activation_function_output))
--
GitLab