Prediction layer name.

rameshvarun · Jun 13, 2017 · 297fe0e · 297fe0e
1 parent 381a560
commit 297fe0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -73,7 +73,7 @@ def create_model(keep_prob=0.6):
     model.add(Dropout(drop_out))
     model.add(Dense(10, activation='relu'))
     model.add(Dropout(drop_out))
-    model.add(Dense(OUT_SHAPE, activation='softsign'))
+    model.add(Dense(OUT_SHAPE, activation='softsign', name="predictions"))
 
     return model