Fix constraint tests
This commit is contained in:
parent
2805413653
commit
bba5379305
@ -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
|
||||
|
@ -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'))
|
||||
|
Loading…
Reference in New Issue
Block a user