Fix constraint tests
This commit is contained in:
parent
2805413653
commit
bba5379305
@ -24,7 +24,7 @@ class NonNeg(Constraint):
|
|||||||
|
|
||||||
class UnitNorm(Constraint):
|
class UnitNorm(Constraint):
|
||||||
def __call__(self, p):
|
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
|
identity = Constraint
|
||||||
maxnorm = MaxNorm
|
maxnorm = MaxNorm
|
||||||
|
@ -14,7 +14,7 @@ class TestConcatenation(unittest.TestCase):
|
|||||||
|
|
||||||
def test_unitnorm_constraint(self):
|
def test_unitnorm_constraint(self):
|
||||||
lookup = Sequential()
|
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(Flatten())
|
||||||
lookup.add(Dense(2, 1))
|
lookup.add(Dense(2, 1))
|
||||||
lookup.add(Activation('sigmoid'))
|
lookup.add(Activation('sigmoid'))
|
||||||
|
Loading…
Reference in New Issue
Block a user