From 297fe0e3bc3ff44c78443d1e5e6a38e53f46e650 Mon Sep 17 00:00:00 2001 From: Varun Ramesh Date: Mon, 12 Jun 2017 20:28:02 -0700 Subject: [PATCH] Prediction layer name. --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index ead43a6..089cd54 100644 --- 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