updated mapping test to pass gradient checks

This commit is contained in:
msbauer 2017-10-12 11:56:20 +02:00 committed by GitHub
parent 42627f4030
commit ef14c7a7eb

View file

@ -46,7 +46,7 @@ class MappingTests(unittest.TestCase):
def test_mlpextmapping(self):
for activation in ['tanh', 'relu', 'sigmoid']:
mapping = GPy.mappings.MLPext(input_dim=3, hidden_dims=[5,5,5], output_dim=2, activation=activation)
mapping = GPy.mappings.MLPext(input_dim=3, hidden_dims=[5,5], output_dim=2, activation=activation)
X = np.random.randn(100,3)
self.assertTrue(MappingGradChecker(mapping, X).checkgrad())