mirror of https://github.com/explosion/spaCy.git
fix parikh entailment test methods bug with settings
This commit is contained in:
parent
89df91846c
commit
a592075720
|
@ -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])
|
||||||
|
|
Loading…
Reference in New Issue