fix parikh entailment test methods bug with settings

This commit is contained in:
Chen Buskilla 2016-11-13 14:53:55 +02:00
parent 89df91846c
commit a592075720
1 changed files with 2 additions and 2 deletions

View File

@ -199,7 +199,7 @@ class _GlobalSumPooling1D(Layer):
def test_build_model(): def test_build_model():
vectors = numpy.ndarray((100, 8), dtype='float32') vectors = numpy.ndarray((100, 8), dtype='float32')
shape = (10, 16, 3) shape = (10, 16, 3)
settings = {'lr': 0.001, 'dropout': 0.2} settings = {'lr': 0.001, 'dropout': 0.2, 'gru_encode':True}
model = build_model(vectors, shape, settings) model = build_model(vectors, shape, settings)
@ -220,7 +220,7 @@ def test_fit_model():
vectors = numpy.ndarray((100, 8), dtype='float32') vectors = numpy.ndarray((100, 8), dtype='float32')
shape = (10, 16, 3) shape = (10, 16, 3)
settings = {'lr': 0.001, 'dropout': 0.2} settings = {'lr': 0.001, 'dropout': 0.2, 'gru_encode':True}
model = build_model(vectors, shape, settings) model = build_model(vectors, shape, settings)
train_X = _generate_X(20, shape[0], vectors.shape[1]) train_X = _generate_X(20, shape[0], vectors.shape[1])