diff options
Diffstat (limited to 'imago/engine/keras/convNeuralNetwork.py')
-rw-r--r-- | imago/engine/keras/convNeuralNetwork.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/imago/engine/keras/convNeuralNetwork.py b/imago/engine/keras/convNeuralNetwork.py index 9d97586..638e2fe 100644 --- a/imago/engine/keras/convNeuralNetwork.py +++ b/imago/engine/keras/convNeuralNetwork.py @@ -38,7 +38,7 @@ class ConvNeuralNetwork(NeuralNetwork): ), Flatten(), Dense( - units=81, + units=82, activation='softmax' ), ]) |