lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix constraint tests

Browse Source
This commit is contained in:
fchollet 2015-07-02 17:47:12 -07:00
parent 2805413653
commit bba5379305
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
keras/constraints.py
View File

@ -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

2
tests/auto/test_embeddings.py
View File

@ -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'))