1 files changed, 1 insertions, 2 deletions
diff --git a/lenet.py b/lenet.py
index 4950fe9..6bea8dd 100644
--- a/lenet.py
+++ b/lenet.py
@@ -64,8 +64,7 @@ def get_lenet(shape):
   
   model.add(Dense(units=120, activation='relu'))
   model.add(Dense(units=84, activation='relu'))
-  #model.add(Dense(units=10, activation = 'softmax'))
-  model.add(Dense(units=10, activation = 'relu'))
+  model.add(Dense(units=10, activation = 'softmax'))
   return model
 
 def get_lenet_icp(shape):