Update zth_train.py

Timthony · Jun 2, 2019 · 3e9353f · 3e9353f
1 parent 1c9f8e7
commit 3e9353f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zth_train.py b/zth_train.py
@@ -78,7 +78,7 @@ def build_model(keep_prob):
     #model.add(Dense(500, activation='elu'))
     model.add(Dense(250, activation='elu'))
     #model.add(Dense(50, activation='elu'))
-    model.add(Dense(5, ACTIVATION='softmax'))
+    model.add(Dense(5, activation='softmax'))
     model.summary()
 
     return model