diff --git a/toolkit.py b/toolkit.py index 3896daf4f9bf9dcbe8cd74d5244e7e16b07b5d85..7f45570bae5750c6ad447c3fcbd60e6d7f2e557d 100755 --- a/toolkit.py +++ b/toolkit.py @@ -915,9 +915,13 @@ class ClassificationProject(object): elif mode == "skip_activation": # output before applying activation function # (after weighted sum + bias of last hidden layer) + if isinstance(self.model.input, list): + feed_dict={tuple(self.model.input) : x} + else: + feed_dict={self.model.input : x} return K.get_session().run( self.model.output.op.inputs[0], - feed_dict={self.model.input : x} + feed_dict=feed_dict ) else: raise ValueError("Unknown mode {}".format(mode))