diff --git a/keras/constraints.py b/keras/constraints.py index 13df3217a..a1164126f 100644 --- a/keras/constraints.py +++ b/keras/constraints.py @@ -24,7 +24,7 @@ class NonNeg(Constraint): class UnitNorm(Constraint): def __call__(self, p): - return e / T.sqrt(T.sum(e**2, axis=-1, keepdims=True)) + return p / T.sqrt(T.sum(p**2, axis=-1, keepdims=True)) identity = Constraint maxnorm = MaxNorm diff --git a/tests/auto/test_embeddings.py b/tests/auto/test_embeddings.py index 7b8308eeb..f4c22e8ba 100644 --- a/tests/auto/test_embeddings.py +++ b/tests/auto/test_embeddings.py @@ -14,7 +14,7 @@ class TestConcatenation(unittest.TestCase): def test_unitnorm_constraint(self): lookup = Sequential() - lookup.add(Embedding(3, 2, weights=[self.W1], W_constraint=unitnorm)) + lookup.add(Embedding(3, 2, weights=[self.W1], W_constraint=unitnorm())) lookup.add(Flatten()) lookup.add(Dense(2, 1)) lookup.add(Activation('sigmoid'))