From 67c9055c066b04bcfc66fe9f68026eb4ccb2704c Mon Sep 17 00:00:00 2001 From: guilherme-pombo Date: Tue, 13 Sep 2016 16:32:09 +0100 Subject: [PATCH] Fixed the eval_sick so that it works with the new Keras version --- eval_sick.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eval_sick.py b/eval_sick.py index 9825710..ddba020 100644 --- a/eval_sick.py +++ b/eval_sick.py @@ -70,7 +70,7 @@ def prepare_model(ninputs=9600, nclass=5): Set up and compile the model architecture (Logistic regression) """ lrmodel = Sequential() - lrmodel.add(Dense(ninputs, nclass)) + lrmodel.add(Dense(nclass, input_dim=ninputs)) lrmodel.add(Activation('softmax')) lrmodel.compile(loss='categorical_crossentropy', optimizer='adam') return lrmodel