mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-06-02 14:45:15 +02:00
updated mapping test to pass gradient checks
This commit is contained in:
parent
42627f4030
commit
ef14c7a7eb
1 changed files with 1 additions and 1 deletions
|
|
@ -46,7 +46,7 @@ class MappingTests(unittest.TestCase):
|
||||||
|
|
||||||
def test_mlpextmapping(self):
|
def test_mlpextmapping(self):
|
||||||
for activation in ['tanh', 'relu', 'sigmoid']:
|
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)
|
X = np.random.randn(100,3)
|
||||||
self.assertTrue(MappingGradChecker(mapping, X).checkgrad())
|
self.assertTrue(MappingGradChecker(mapping, X).checkgrad())
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue